Update app.py

app.py

@@ -4,37 +4,104 @@
 
 import anthropic
 import gradio as gr
+import os
 
+# Initialize the Anthropic client
 client = anthropic.Anthropic()
 
 def predict(message, history):
-    keys_to_keep = ["role", "content"]
-    history = [{k: d[k] for k in keys_to_keep if k in d} for d in history]
-    history.append({"role": "user", "content": message})
-    if len(history) > 20:
-        history.append({"role": "user", "content": "DONE"})
-    output = client.messages.create(
-        messages=history,  
-        model="claude-3-5-sonnet-20241022",
-        max_tokens=1000,
-        system="You are guessing an object that the user is thinking of. You can ask 10 yes/no questions. Keep asking questions until the user says DONE"
-    )
-    return {
-        "role": "assistant",
-        "content": output.content[0].text,  
-        "options": [{"value": "Yes"}, {"value": "No"}]
-    }
+    try:
+        # Convert history to the format expected by Anthropic API
+        anthropic_history = []
+        for msg in history:
+            if isinstance(msg, dict):
+                anthropic_history.append(msg)
+            else:
+                # Handle tuple format if it exists
+                role = "user" if msg[0] else "assistant"
+                anthropic_history.append({"role": role, "content": msg[1]})
+        
+        # Add the current message
+        anthropic_history.append({"role": "user", "content": message})
+        
+        # Count questions asked (only count assistant messages that are questions)
+        question_count = sum(1 for msg in anthropic_history 
+                           if msg["role"] == "assistant" and "?" in msg["content"])
+        
+        # Check if we've reached the limit or user said "DONE"
+        if question_count >= 10 or message.upper() == "DONE":
+            system_prompt = """You are playing a guessing game. The user was thinking of something and you've been asking yes/no questions to guess it. Now either you've used up your 10 questions or the user said DONE. Make your final guess about what they were thinking of, or ask them to reveal the answer if you're not sure."""
+        else:
+            system_prompt = f"""You are playing a guessing game where you try to guess what the user is thinking of (a person, place, or thing). You can ask up to 10 yes/no questions. You have asked {question_count} questions so far. Ask strategic yes/no questions to narrow down the possibilities. Keep your questions concise and clear."""
+        
+        # Make API call to Anthropic
+        response = client.messages.create(
+            model="claude-3-5-sonnet-20241022",
+            max_tokens=1000,
+            system=system_prompt,
+            messages=anthropic_history
+        )
+        
+        # Return just the text content - Gradio will handle the formatting
+        return response.content[0].text
+        
+    except Exception as e:
+        return f"Sorry, I encountered an error: {str(e)}"
+
+# Custom CSS for better styling
+custom_css = """
+#chatbot {
+    height: 500px;
+}
+.message {
+    font-size: 16px;
+}
+"""
 
 placeholder = """
-<center><h1>10 Questions</h1><br>Think of a person, place, or thing. I'll ask you 10 yes/no questions to try and guess it.
+<center>
+<h1>🎯 10 Questions Game</h1>
+<p>Think of a person, place, or thing. I'll ask you up to 10 yes/no questions to try and guess it!</p>
+<p><strong>Instructions:</strong></p>
+<ul style="text-align: left; display: inline-block;">
+<li>Think of something specific (person, place, or thing)</li>
+<li>Answer my questions with "Yes" or "No"</li>
+<li>Type "DONE" if you want me to make my final guess early</li>
+<li>Have fun! 🎉</li>
+</ul>
 </center>
 """
 
+# Create the Gradio interface
 demo = gr.ChatInterface(
     predict,
-    examples=["Start!"],
-    chatbot=gr.Chatbot(placeholder=placeholder),
-    type="messages"
+    title="10 Questions Guessing Game",
+    examples=["Let's start!", "I'm ready!", "Begin the game!"],
+    chatbot=gr.Chatbot(
+        placeholder=placeholder,
+        height=500,
+        show_copy_button=True
+    ),
+    textbox=gr.Textbox(
+        placeholder="Answer 'Yes' or 'No' to my questions, or type 'DONE' to finish early...",
+        container=False,
+        scale=7
+    ),
+    type="messages",
+    css=custom_css,
+    theme=gr.themes.Soft()
 )
 
-demo.launch()
+if __name__ == "__main__":
+    # Check if API key is set
+    if not os.getenv("ANTHROPIC_API_KEY"):
+        print("⚠️  Warning: ANTHROPIC_API_KEY environment variable not set!")
+        print("Please set it with: export ANTHROPIC_API_KEY='your-api-key'")
+    
+    print("🚀 Starting 10 Questions Game...")
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        debug=True
+    )