Update second version(2.0).py

second version(2.0).py

@@ -8,7 +8,13 @@ f4 = ["ap13", "ap14", "ap15", "ap16"]
 building = [f1 , f2, f3, f4]
 bombs = []
 
-bq = random.randint(1,10)
+bq = int(input("How many bombs do you desire? "))
+print("please answer the following with yes or no")
+twoin1 = str(input("Do you want more then one bomb in an apartment? "))
+
+twoin1 =twoin1.lower()
+#I made it possible for the user to choose how many bombs he wants to be generated, and if he wants
+#to have the possibility for there to be more then one bomb in an apartment
 g = 1
 
 for g in range(bq):
@@ -16,13 +22,12 @@ for g in range(bq):
     b = random.randint(0, 3)
     bl = building[a][b]
     bombs.append(bl)
-    bombs2 = set(dict.fromkeys(bombs))#I was testing with lists , then had to make a dictionary to remove duplicates,
+        # I was testing with lists , then had to make a dictionary to remove duplicates,
         # but for the next version i will use SETS
-    size = len(bombs2)
-    #print(bl, a, b, bq)
-print(bombs)
-print(bombs2)
 reroll = []
+if twoin1 == "no":
+    bombs2 = set(dict.fromkeys(bombs))
+    size = len(bombs2)
     while size != bq:
         r = bq - size
         for g in range(r):
@@ -42,9 +47,10 @@ while size != bq:
         else:
             continue
         break
+else:pass
 #I was able to make this verification to work, thus my code can be true to the bomb quantity that is randomly generated, now i need to make the floor verification
 #and make it so the user determines how many bombs will be generated
-exit()
+exit() #I will now star to program the next section of my challenge and i will upload it as a 3.0
 #I put this exit() to do some testing so please ignore what comes in the next blocks of code since i am testing a way
 # to generate random bombs and avoid duplicates
 #After achieving this i will make thhe user input so you may choose how many bombs you want, and use a bollean to