Update app.py

app.py

@@ -1,96 +1,106 @@
 import gradio as gr
-from self_learning_bot import UnrestrictedChatbot
+from self_learning_bot import EnhancedLearningBot
 import os
+import time
 
-# Initialize bot with persistent storage
-state_path = "/tmp/chatbot_state.json"
-bot = UnrestrictedChatbot(state_file=state_path)
+# Initialize the enhanced bot
+bot = EnhancedLearningBot()
 
-def respond(message, history):
-    bot.learn_from_feedback("", 1.0)  # Learn from successful responses
+def respond(message, history, enable_search):
+    # Get response with optional web search
+    response, search_context = bot.chat(
+        message, 
+        use_web_search=enable_search,
+        conversation_history=history
+    )
     
-    # Get bot response
-    response = bot.chat(message)
+    # Learn from this interaction
+    bot.learn_from_interaction(message, response, search_context)
     
-    # Show learning stats
+    # Display learning stats and search info
     stats = bot.get_learning_stats()
-    learning_info = f"🤖 Patterns: {stats['patterns']} | Memory: {stats['memory_size']} | Score: {stats['avg_score']:.2f}"
+    search_info = "🔍 Web Search Used" if search_context else "💭 From Memory"
+    learning_info = f"{search_info} | Patterns: {stats['patterns']} | Score: {stats['avg_score']:.2f}"
     
     if history is None:
         history = []
-    history.append((message, f"{response}\n\n{learning_info}"))
+    full_response = f"{response}\n\n{learning_info}"
+    history.append((message, full_response))
     return history, history
 
 def add_feedback(feedback, history):
     if feedback and history:
-        # Learn from user feedback (1-5 scale)
         try:
             score = float(feedback)
             if 1 <= score <= 5:
-                bot.learn_from_feedback(history[-1][0], score / 5.0)  # Convert to 0-1 scale
+                bot.learn_from_feedback(history[-1][0], score / 5.0)
                 return f"✅ Learned from feedback: {score}/5"
         except:
             pass
-    return "❌ Please enter a number 1-5"
+    return "❌ Please enter 1-5"
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="Phoenix AI - Web Enhanced") as demo:
     gr.Markdown("""
-    # 🚀 Phoenix AI (Real Self-Learning Chatbot)
-    *This bot actually learns from every conversation and improves over time!*
+    # 🚀 Phoenix AI - Web Enhanced
+    **Now with real-time web search and continuous learning!**
     """)
     
     chatbot = gr.Chatbot(height=400, label="Learning Chat")
     state = gr.State([])
     
     with gr.Row():
-        msg = gr.Textbox(
-            label="Your message",
-            placeholder="Type anything... the bot learns from your conversations!",
-            scale=4
-        )
-        send_btn = gr.Button("Send", scale=1)
+        with gr.Column(scale=4):
+            msg = gr.Textbox(
+                label="Your message",
+                placeholder="Ask me anything... I can search the web for latest information!",
+                container=False
+            )
+        with gr.Column(scale=1):
+            search_toggle = gr.Checkbox(label="🔍 Enable Web Search", value=True)
+            send_btn = gr.Button("Send", variant="primary")
     
     with gr.Row():
         feedback = gr.Textbox(
             label="Rate last response (1-5)",
-            placeholder="Help me learn! Rate 1(bad) to 5(great)",
+            placeholder="Help me learn! Rate 1-5",
             scale=3
         )
         learn_btn = gr.Button("Submit Feedback", scale=1)
-        reset_btn = gr.Button("Reset Chat", scale=1)
+        clear_btn = gr.Button("Clear Chat", scale=1)
     
     learning_stats = gr.Textbox(
         label="Learning Statistics",
-        interactive=False,
-        value="Bot will show learning progress here..."
+        value="Ready to learn and search...",
+        interactive=False
     )
     
-    # Update stats
+    # Update stats display
     def update_stats():
         stats = bot.get_learning_stats()
-        return f"📊 Learned Patterns: {stats['patterns']} | Conversations: {stats['memory_size']} | Avg Score: {stats['avg_score']:.2f} | Recent Rewards: {stats['recent_rewards']}"
+        return f"📊 Learned: {stats['patterns']} patterns | Memory: {stats['memory_size']} | Recent Success: {stats['recent_rewards']}/10"
     
-    # Main interactions
-    def user_message(message, history):
+    # Main interaction flow
+    def user_message(message, history, search_toggle):
         if not message.strip():
             return "", history
-        new_history = respond(message, history or [])
+        new_history = respond(message, history or [], search_toggle)
         return "", new_history[0], update_stats()
     
-    msg.submit(user_message, [msg, state], [msg, chatbot, learning_stats])
-    send_btn.click(user_message, [msg, state], [msg, chatbot, learning_stats])
+    # Event handlers
+    msg.submit(user_message, [msg, state, search_toggle], [msg, chatbot, learning_stats])
+    send_btn.click(user_message, [msg, state, search_toggle], [msg, chatbot, learning_stats])
     
     learn_btn.click(
-        add_feedback, 
-        [feedback, chatbot], 
+        add_feedback,
+        [feedback, chatbot],
         [feedback]
     ).then(update_stats, None, [learning_stats])
     
-    reset_btn.click(
-        lambda: ([], [], "Chat reset!"), 
-        None, 
+    clear_btn.click(
+        lambda: ([], [], "Chat cleared!"),
+        None,
         [chatbot, state, learning_stats]
     )
 
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    demo.launch(share=True, show_error=True)