Update app.py

app.py

@@ -43,9 +43,8 @@ def flip_coin():
         is_heads = random.random() < coin['winrate']
         if is_heads:
             game_state['balance'] += coin['value']
-            # Apply coin bonus if it exists
-            if coin['bonus'] != 'none':
-                apply_bonus(coin['bonus'])
+            # Apply coin bonus
+            game_state['balance'] += coin['bonus']
         game_state['flips_left'] -= 1
         
         return jsonify({
@@ -61,7 +60,7 @@ def buy_coin():
     if index < len(coins) and game_state['balance'] >= coins[index]['price'] and index != game_state['current_coin']:
         game_state['balance'] -= coins[index]['price']
         game_state['current_coin'] = index
-        return jsonify({'success': True, 'balance': game_state['balance']})
+        return jsonify({'success': True, 'balance': game_state['balance'], 'current_coin': index})
     return jsonify({'success': False})
 
 @app.route('/generate_coin', methods=['POST'])
@@ -72,10 +71,10 @@ def generate_coin():
     prompt = """Generate a new coin for a game in JSON format with the following properties:
     name (a creative name for the coin),
     color (as a hex code),
-    price (higher than 10),
-    value (higher than 0.1),
+    price (between 10 and 100),
+    value (between 0.1 and 1.0),
     winrate (between 0.5 and 0.8),
-    bonus (a bonus effect that triggers on successful flips, e.g., '+2 flips_left', '+0.5 balance', 'none')
+    bonus (a flat $ value between 0.01 and 0.1)
     
     Example:
     {
@@ -84,16 +83,16 @@ def generate_coin():
         "price": 15,
         "value": 0.2,
         "winrate": 0.65,
-        "bonus": "+2 flips_left"
+        "bonus": 0.05
     }
     """
 
-    inputs = tokenizer(prompt, return_tensors="pt")
-    with torch.no_grad():
-        outputs = model.generate(**inputs, max_length=300)
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
     try:
+        inputs = tokenizer(prompt, return_tensors="pt")
+        with torch.no_grad():
+            outputs = model.generate(**inputs, max_length=300)
+        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
         new_coin = json.loads(response)
         if validate_coin(new_coin):
             coins.append(new_coin)
@@ -103,8 +102,9 @@ def generate_coin():
             return jsonify({'success': True, 'coin': new_coin, 'balance': game_state['balance']})
         else:
             return jsonify({'success': False, 'error': 'Invalid coin format'})
-    except json.JSONDecodeError:
-        return jsonify({'success': False, 'error': 'Invalid JSON format'})
+    except Exception as e:
+        # Refund the player if coin generation fails
+        return jsonify({'success': False, 'error': str(e), 'balance': game_state['balance']})
 
 @app.route('/game_over', methods=['POST'])
 def game_over():
@@ -129,18 +129,10 @@ def validate_coin(coin):
     return all(key in coin for key in required_keys) and \
            isinstance(coin['name'], str) and \
            isinstance(coin['color'], str) and coin['color'].startswith('#') and \
-           isinstance(coin['price'], (int, float)) and coin['price'] > 10 and \
-           isinstance(coin['value'], (int, float)) and coin['value'] > 0.1 and \
+           isinstance(coin['price'], (int, float)) and 10 <= coin['price'] <= 100 and \
+           isinstance(coin['value'], (int, float)) and 0.1 <= coin['value'] <= 1.0 and \
            isinstance(coin['winrate'], (int, float)) and 0.5 <= coin['winrate'] <= 0.8 and \
-           isinstance(coin['bonus'], str)
-
-def apply_bonus(bonus):
-    if bonus.startswith('+'):
-        value, attribute = bonus.split()[0][1:], bonus.split()[1]
-        if attribute == 'flips_left':
-            game_state['flips_left'] += int(value)
-        elif attribute == 'balance':
-            game_state['balance'] += float(value)
+           isinstance(coin['bonus'], (int, float)) and 0.01 <= coin['bonus'] <= 0.1
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)