Update app.py

app.py

@@ -49,7 +49,7 @@ class ChatbotManager:
         return f"✅ Custom data integrated ({len(data_text)} characters)"
     
     def generate_response(self, user_input: str, history: List[Tuple[str, str]]) -> Tuple[str, List[Tuple[str, str]]]:
-        """Generate response using the selected LLM model"""
+        """Generate response using the selected LLM model with user and AI icons"""
         if not self.current_api_key:
             return "❌ Please set your API key first!", history
         
@@ -60,13 +60,13 @@ class ChatbotManager:
             # Prepare conversation context
             messages = [{"role": "system", "content": self.system_prompt}]
             
-            # Add conversation history
+            # Add conversation history with icons
             for user_msg, assistant_msg in history:
-                messages.append({"role": "user", "content": user_msg})
-                messages.append({"role": "assistant", "content": assistant_msg})
+                messages.append({"role": "user", "content": f"👤 {user_msg}"})
+                messages.append({"role": "assistant", "content": f"🤖 {assistant_msg}"})
             
-            # Add current user input
-            messages.append({"role": "user", "content": user_input})
+            # Add current user input with icon
+            messages.append({"role": "user", "content": f"👤 {user_input}"})
             
             # Generate response
             response = openai.ChatCompletion.create(
@@ -79,43 +79,44 @@ class ChatbotManager:
             )
             
             assistant_response = response.choices[0].message.content.strip()
+            formatted_response = f"🤖 {assistant_response}"  # Prepend AI icon to response
             
             # Update history
-            history.append((user_input, assistant_response))
+            history.append((user_input, assistant_response))  # Store without icons for history
             
-            return assistant_response, history
+            return formatted_response, history
             
         except Exception as e:
             error_msg = f"❌ Error generating response: {str(e)}"
             return error_msg, history
-    
-    def clear_conversation(self) -> Tuple[str, List[Tuple[str, str]]]:
-        """Clear conversation history"""
-        self.conversation_history = []
-        return "", []
-    
-    def export_conversation(self, history: List[Tuple[str, str]]) -> str:
-        """Export conversation history to JSON format"""
-        if not history:
-            return "No conversation to export"
         
-        export_data = {
-            "timestamp": datetime.now().isoformat(),
-            "model": self.current_model,
-            "conversation": [
-                {"user": user_msg, "assistant": assistant_msg}
-                for user_msg, assistant_msg in history
-            ]
-        }
+        def clear_conversation(self) -> Tuple[str, List[Tuple[str, str]]]:
+            """Clear conversation history"""
+            self.conversation_history = []
+            return "", []
         
-        filename = f"conversation_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
-        
-        try:
-            with open(filename, 'w', encoding='utf-8') as f:
-                json.dump(export_data, f, indent=2, ensure_ascii=False)
-            return f"✅ Conversation exported to {filename}"
-        except Exception as e:
-            return f"❌ Export failed: {str(e)}"
+        def export_conversation(self, history: List[Tuple[str, str]]) -> str:
+            """Export conversation history to JSON format"""
+            if not history:
+                return "No conversation to export"
+            
+            export_data = {
+                "timestamp": datetime.now().isoformat(),
+                "model": self.current_model,
+                "conversation": [
+                    {"user": user_msg, "assistant": assistant_msg}
+                    for user_msg, assistant_msg in history
+                ]
+            }
+            
+            filename = f"conversation_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
+            
+            try:
+                with open(filename, 'w', encoding='utf-8') as f:
+                    json.dump(export_data, f, indent=2, ensure_ascii=False)
+                return f"✅ Conversation exported to {filename}"
+            except Exception as e:
+                return f"❌ Export failed: {str(e)}"
 
 # Initialize chatbot manager
 chatbot = ChatbotManager()