Update app.py

app.py

@@ -1,15 +1,3 @@
-import gradio as gr
-from huggingface_hub import InferenceClient
-from typing import List, Tuple
-
-# Initialize the InferenceClient with the model you want to use
-client = InferenceClient("microsoft/phi-4")
-
-# Define the updated system message focused on programming and coding
-SYSTEM_MESSAGE = """
-You're an advanced AI assistant specializing in computer programming and coding. Your role is to assist users with programming-related queries by providing helpful, clear, and concise explanations and solutions. You can help with debugging, code optimization, algorithm design, and provide advice on best practices. You should aim to respond in a way that's both friendly and informative, maintaining a conversational tone. Feel free to offer code examples, explain concepts in detail, and guide users through coding challenges.
-"""
-
 def generate_response(
     user_input: str, 
     history: List[Tuple[str, str]], 
@@ -45,31 +33,18 @@ def generate_response(
             temperature=temperature,
             top_p=top_p,
         ):
-            if 'choices' in msg and len(msg['choices']) > 0:
+            # Check if 'choices' is present and non-empty
+            if msg and 'choices' in msg and msg['choices']:
+                # Ensure the 'delta' and 'content' properties exist before using them
                 token = msg['choices'][0].get('delta', {}).get('content', '')
                 if token:
                     response += token
-        return response
+            else:
+                # Handle unexpected response format or empty choices
+                print("Warning: Unexpected response format or empty 'choices'.")
+                break
+        return response or "Sorry, I couldn't generate a response. Please try again."
 
     except Exception as e:
         print(f"An error occurred: {e}")
         return "Error: An unexpected error occurred while processing your request."
-
-# Define the Gradio Interface
-iface = gr.Interface(
-    fn=generate_response,
-    inputs=[
-        gr.Textbox(lines=2, label="Your Message 💬"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens 🔢"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature 🌡️"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p 🎯"),
-        gr.Chatbot(label="Programming Assistant 💻")
-    ],
-    outputs=[gr.Textbox(label="AI Response 🤖")],
-    title="CodeChat: Programming Assistant 💬",
-    description="Ask programming-related questions and get clear, helpful, and concise coding advice and solutions. Whether you're debugging, optimizing, or learning new concepts, I'm here to help! 👨‍💻",
-)
-
-# Launch the interface
-if __name__ == "__main__":
-    iface.launch()