Update app.py

app.py

@@ -22,11 +22,13 @@ class ChatbotManager:
             return "❌ Please enter a valid API key"
         
         self.current_api_key = api_key.strip()
-        openai.api_key = self.current_api_key
+        
+        # Create OpenAI client with the new syntax
+        self.client = openai.OpenAI(api_key=self.current_api_key)
         
         # Test the API key
         try:
-            openai.Model.list()
+            self.client.models.list()
             return "✅ API key validated successfully!"
         except Exception as e:
             return f"❌ Invalid API key: {str(e)}"
@@ -44,12 +46,13 @@ class ChatbotManager:
         if not data_text.strip():
             return "No custom data provided"
         
-        # Add custom data to system prompt
-        self.system_prompt += f"\n\nAdditional Context:\n{data_text}"
+        # Reset system prompt to avoid accumulation
+        base_prompt = "You are a helpful AI assistant. Respond in a friendly and informative manner."
+        self.system_prompt = base_prompt + f"\n\nAdditional Context:\n{data_text}"
         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 with user and AI icons"""
+        """Generate response using the selected LLM model"""
         if not self.current_api_key:
             return "❌ Please set your API key first!", history
         
@@ -60,16 +63,16 @@ class ChatbotManager:
             # Prepare conversation context
             messages = [{"role": "system", "content": self.system_prompt}]
             
-            # Add conversation history with icons
+            # Add conversation history
             for user_msg, assistant_msg in history:
-                messages.append({"role": "user", "content": f"👤 {user_msg}"})
-                messages.append({"role": "assistant", "content": f"🤖 {assistant_msg}"})
+                messages.append({"role": "user", "content": user_msg})
+                messages.append({"role": "assistant", "content": assistant_msg})
             
-            # Add current user input with icon
-            messages.append({"role": "user", "content": f"👤 {user_input}"})
+            # Add current user input
+            messages.append({"role": "user", "content": user_input})
             
-            # Generate response
-            response = openai.ChatCompletion.create(
+            # Generate response using new OpenAI client syntax
+            response = self.client.chat.completions.create(
                 model=self.current_model,
                 messages=messages,
                 max_tokens=self.max_tokens,
@@ -79,44 +82,43 @@ 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))  # Store without icons for history
+            history.append((user_input, assistant_response))
             
-            return formatted_response, history
+            return assistant_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"
         
-        def clear_conversation(self) -> Tuple[str, List[Tuple[str, str]]]:
-            """Clear conversation history"""
-            self.conversation_history = []
-            return "", []
+        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 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)}"
+        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()
@@ -151,7 +153,9 @@ def create_interface():
                         height=400,
                         show_label=True,
                         avatar_images=("👤", "🤖"),
-                        show_copy_button=True
+                        show_copy_button=True,
+                        bubble_full_width=False,
+                        show_share_button=True
                     )
                     
                     with gr.Row():
@@ -159,19 +163,20 @@ def create_interface():
                             placeholder="Type your message here...",
                             scale=4,
                             show_label=False,
-                            
+                            container=False
                         )
-                        send_btn = gr.Button("Send", variant="primary", scale=1)
+                        send_btn = gr.Button("📤 Send", variant="primary", scale=1)
                     
                     with gr.Row():
-                        clear_btn = gr.Button("Clear Chat", variant="secondary")
-                        export_btn = gr.Button("Export Chat", variant="secondary")
+                        clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
+                        export_btn = gr.Button("📥 Export Chat", variant="secondary")
+                        regenerate_btn = gr.Button("🔄 Regenerate", variant="secondary")
                 
                 with gr.Column(scale=1):
                     gr.Markdown("### 🔧 Quick Settings")
                     
                     api_key_input = gr.Textbox(
-                        label="OpenAI API Key",
+                        label="🔑 OpenAI API Key",
                         placeholder="sk-...",
                         type="password"
                     )
@@ -184,15 +189,15 @@ def create_interface():
                     model_dropdown = gr.Dropdown(
                         choices=AVAILABLE_MODELS,
                         value="gpt-3.5-turbo",
-                        label="Model"
+                        label="🤖 Model"
                     )
                     
                     max_tokens_slider = gr.Slider(
                         minimum=50,
-                        maximum=500,
+                        maximum=2000,
                         value=150,
                         step=10,
-                        label="Max Tokens"
+                        label="📏 Max Tokens"
                     )
                     
                     temperature_slider = gr.Slider(
@@ -200,11 +205,20 @@ def create_interface():
                         maximum=1.0,
                         value=0.7,
                         step=0.1,
-                        label="Temperature"
+                        label="🌡️ Temperature"
+                    )
+                    
+                    # Live settings display
+                    gr.Markdown("### 📊 Current Settings")
+                    current_settings = gr.Textbox(
+                        value="Model: gpt-3.5-turbo\nTokens: 150\nTemp: 0.7",
+                        label="Active Configuration",
+                        interactive=False,
+                        lines=3
                     )
         
         with gr.Tab("⚙️ Advanced Settings"):
-            gr.Markdown("### System Prompt Configuration")
+            gr.Markdown("### 🎯 System Prompt Configuration")
             system_prompt_input = gr.Textbox(
                 label="System Prompt",
                 value="You are a helpful AI assistant. Respond in a friendly and informative manner.",
@@ -212,7 +226,7 @@ def create_interface():
                 placeholder="Enter custom system prompt..."
             )
             
-            gr.Markdown("### 📊 Custom Data Integration")
+            gr.Markdown("### 📚 Custom Data Integration")
             custom_data_input = gr.Textbox(
                 label="Custom Training Data",
                 lines=10,
@@ -220,13 +234,22 @@ def create_interface():
             )
             
             with gr.Row():
-                update_settings_btn = gr.Button("Update Settings", variant="primary")
-                integrate_data_btn = gr.Button("Integrate Custom Data", variant="secondary")
+                update_settings_btn = gr.Button("✅ Update Settings", variant="primary")
+                integrate_data_btn = gr.Button("📊 Integrate Custom Data", variant="secondary")
+                reset_prompt_btn = gr.Button("🔄 Reset to Default", variant="secondary")
             
             settings_status = gr.Textbox(
                 label="Settings Status",
                 interactive=False
             )
+            
+            # Preset system prompts
+            gr.Markdown("### 🎭 Preset System Prompts")
+            with gr.Row():
+                preset_customer_support = gr.Button("👥 Customer Support", variant="secondary")
+                preset_tutor = gr.Button("🎓 Educational Tutor", variant="secondary")
+                preset_creative = gr.Button("✨ Creative Assistant", variant="secondary")
+                preset_technical = gr.Button("🔧 Technical Writer", variant="secondary")
         
         with gr.Tab("📋 Usage Guide"):
             gr.Markdown("""
@@ -234,22 +257,25 @@ def create_interface():
             
             ### 1. **Set Up API Key**
             - Obtain an OpenAI API key from [OpenAI Platform](https://platform.openai.com/)
-            - Enter your API key in the "OpenAI API Key" field
+            - Enter your API key in the "🔑 OpenAI API Key" field
             - Wait for the green checkmark confirmation
             
             ### 2. **Configure Settings**
             - **Model**: Choose from available GPT models
-            - **Max Tokens**: Control response length (50-500)
+            - **Max Tokens**: Control response length (50-2000)
             - **Temperature**: Adjust creativity (0.0 = focused, 1.0 = creative)
             
             ### 3. **Advanced Customization**
             - **System Prompt**: Define the AI's personality and behavior
             - **Custom Data**: Add domain-specific information or FAQs
+            - **Presets**: Use pre-configured prompts for common use cases
             
             ### 4. **Chat Features**
             - Type messages and get intelligent responses
             - Clear conversation history anytime
             - Export chat history as JSON
+            - Regenerate the last response
+            - Copy responses using the copy button
             
             ## 🛠️ Technical Features
             
@@ -258,6 +284,7 @@ def create_interface():
             - **Custom data integration**: Enhance responses with your own data
             - **Export functionality**: Save conversations for later analysis
             - **Real-time validation**: API key and settings verification
+            - **Visual indicators**: User (👤) and AI (🤖) avatars
             
             ## 💡 Use Cases
             
@@ -265,6 +292,13 @@ def create_interface():
             - **Education**: Build tutoring assistants with custom curriculum
             - **Business**: Develop FAQ bots with company-specific information
             - **Research**: Analyze conversations and response patterns
+            
+            ## 🔧 Troubleshooting
+            
+            - **API Key Issues**: Ensure your key is valid and has credits
+            - **Model Errors**: Some models may not be available in your region
+            - **Long Response Times**: Reduce max tokens or switch to faster models
+            - **Context Limits**: Clear chat history if responses become inconsistent
             """)
         
         # Event handlers
@@ -281,7 +315,8 @@ def create_interface():
         
         def handle_settings_update(model, system_prompt, max_tokens, temperature):
             status = chatbot.update_settings(model, system_prompt, max_tokens, temperature)
-            return status
+            settings_display = f"Model: {model}\nTokens: {max_tokens}\nTemp: {temperature}"
+            return status, settings_display
         
         def handle_data_integration(custom_data):
             status = chatbot.preprocess_data(custom_data)
@@ -293,6 +328,33 @@ def create_interface():
         def handle_export(history):
             return chatbot.export_conversation(history)
         
+        def handle_regenerate(history):
+            if not history:
+                return history
+            
+            # Get the last user message and regenerate response
+            last_user_msg = history[-1][0]
+            history_without_last = history[:-1]
+            
+            response, updated_history = chatbot.generate_response(last_user_msg, history_without_last)
+            return updated_history
+        
+        def update_settings_display(model, max_tokens, temperature):
+            return f"Model: {model}\nTokens: {max_tokens}\nTemp: {temperature}"
+        
+        def reset_prompt():
+            default_prompt = "You are a helpful AI assistant. Respond in a friendly and informative manner."
+            return default_prompt, "✅ System prompt reset to default"
+        
+        def load_preset_prompt(preset_type):
+            presets = {
+                "customer_support": "You are a helpful customer support representative. You are friendly, professional, and knowledgeable. Always try to resolve customer issues and provide clear solutions. If you cannot solve a problem, escalate it politely.",
+                "tutor": "You are an experienced tutor. Explain concepts clearly, use examples, and encourage students when they struggle. Break down complex problems into smaller, manageable steps. Always check for understanding.",
+                "creative": "You are a creative writing assistant who helps with stories, poems, and creative content. Provide constructive feedback, suggest improvements, and inspire creativity while maintaining quality standards.",
+                "technical": "You are a technical writer who creates clear, concise documentation. Use precise language, provide examples when relevant, and structure information logically for developers and technical users."
+            }
+            return presets.get(preset_type, ""), f"✅ Loaded {preset_type.replace('_', ' ').title()} preset"
+        
         # Connect events
         api_key_input.change(
             handle_api_key,
@@ -315,7 +377,7 @@ def create_interface():
         update_settings_btn.click(
             handle_settings_update,
             inputs=[model_dropdown, system_prompt_input, max_tokens_slider, temperature_slider],
-            outputs=[settings_status]
+            outputs=[settings_status, current_settings]
         )
         
         integrate_data_btn.click(
@@ -334,6 +396,46 @@ def create_interface():
             inputs=[chatbot_interface],
             outputs=[settings_status]
         )
+        
+        regenerate_btn.click(
+            handle_regenerate,
+            inputs=[chatbot_interface],
+            outputs=[chatbot_interface]
+        )
+        
+        # Live settings update
+        for component in [model_dropdown, max_tokens_slider, temperature_slider]:
+            component.change(
+                update_settings_display,
+                inputs=[model_dropdown, max_tokens_slider, temperature_slider],
+                outputs=[current_settings]
+            )
+        
+        # Reset and preset buttons
+        reset_prompt_btn.click(
+            reset_prompt,
+            outputs=[system_prompt_input, settings_status]
+        )
+        
+        preset_customer_support.click(
+            lambda: load_preset_prompt("customer_support"),
+            outputs=[system_prompt_input, settings_status]
+        )
+        
+        preset_tutor.click(
+            lambda: load_preset_prompt("tutor"),
+            outputs=[system_prompt_input, settings_status]
+        )
+        
+        preset_creative.click(
+            lambda: load_preset_prompt("creative"),
+            outputs=[system_prompt_input, settings_status]
+        )
+        
+        preset_technical.click(
+            lambda: load_preset_prompt("technical"),
+            outputs=[system_prompt_input, settings_status]
+        )
     
     return demo
 
@@ -357,6 +459,8 @@ def print_setup_instructions():
     
     📂 Files Created:
     - conversation_YYYYMMDD_HHMMSS.json (exported chats)
+    
+    🌐 Access: http://localhost:7860
     """)
 
 if __name__ == "__main__":
@@ -367,5 +471,7 @@ if __name__ == "__main__":
     
     # Launch with custom settings
     demo.launch(
-        share=True            
+        share=True,
+        debug=True,
+        show_error=True
     )