Update app.py

app.py

@@ -12,8 +12,6 @@ class ChatAssistant:
         self.generator = None
         self.tokenizer = None
         self.model = None
-        
-        # Load model on initialization
         self.load_model()
     
     def load_model(self):
@@ -23,7 +21,6 @@ class ChatAssistant:
             
         try:
             model_name = "microsoft/DialoGPT-medium"
-            
             self.tokenizer = AutoTokenizer.from_pretrained(model_name)
             self.model = AutoModelForCausalLM.from_pretrained(
                 model_name,
@@ -56,7 +53,6 @@ class ChatAssistant:
             return "Sorry, I'm having technical difficulties. Please try again later."
         
         try:
-            # Create prompt that encourages complete answers
             prompt = f"""The following is a conversation with an AI assistant. The assistant is helpful, knowledgeable, and provides detailed answers.
 
 User: {query}
@@ -83,12 +79,9 @@ AI:"""
 
     def process_message(self, message: str) -> str:
         """Process user message and generate response"""
-        # Check for name introduction
         name_response = self.get_user_name(message)
         if name_response:
             return name_response
-        
-        # Generate AI response
         return self.generate_response(message)
 
     def get_user_name(self, message):
@@ -110,19 +103,21 @@ AI:"""
 # Initialize the assistant
 assistant = ChatAssistant()
 
-def chat_response(message, history):
+def chat_response(message, chat_history):
     """Generate response for Gradio chat interface"""
     if not message.strip():
-        return "Please enter a message!"
+        return chat_history, ""
     
-    return assistant.process_message(message)
+    bot_message = assistant.process_message(message)
+    chat_history.append((message, bot_message))
+    return chat_history, ""
 
 def greet():
-    return random.choice([
+    return [(None, random.choice([
         "Hello! I'm your AI assistant. How can I help you today?",
         "Hi there! What would you like to know?",
         "Welcome! I'm ready to answer your questions."
-    ])
+    ]))]
 
 # Create Gradio interface
 def create_interface():
@@ -132,7 +127,7 @@ def create_interface():
     ) as iface:
         
         chatbot = gr.Chatbot(
-            value=[(None, greet())],
+            value=greet(),
             height=500,
             label="AI Assistant"
         )
@@ -145,12 +140,15 @@ def create_interface():
             )
             submit_btn = gr.Button("Send", variant="primary")
         
-        msg.submit(chat_response, [msg, chatbot], [chatbot])
-        submit_btn.click(chat_response, [msg, chatbot], [chatbot])
+        clear_btn = gr.Button("Clear Chat")
+        
+        msg.submit(chat_response, [msg, chatbot], [chatbot, msg])
+        submit_btn.click(chat_response, [msg, chatbot], [chatbot, msg])
+        clear_btn.click(lambda: (greet(), ""), outputs=[chatbot, msg])
     
     return iface
 
 # Launch the interface
 if __name__ == "__main__":
     interface = create_interface()
-    interface.launch()
+    interface.launch(share=True)