Update app.py

app.py

@@ -18,18 +18,21 @@ client = Groq(api_key=api_key)
 MODEL_NAME = os.environ.get("MODEL_NAME", "llama3-8b-8192")
 
 # Define a function to handle chat completions
-def get_completion(user_input, chat_history):
+def get_completion(user_input):
     if not user_input.strip():
-        return "", chat_history
+        return "Please enter a valid query."
     
     # Check if the user asks "Who made you?"
     if "who made you" in user_input.lower():
-        return "I was created by Thirumoorthi, a brilliant mind working on AI systems!", chat_history + [("user", user_input), ("assistant", "I was created by Thirumoorthi, a brilliant mind working on AI systems!")]
+        return "I was created by Thirumoorthi, a brilliant mind working on AI systems!"
     
     try:
         completion = client.chat.completions.create(
             model=MODEL_NAME,
-            messages=[{"role": "system", "content": "You are a helpful assistant."}] + [{"role": "user", "content": msg} for role, msg in chat_history] + [("user", user_input)],
+            messages=[
+                {"role": "system", "content": "You are a friendly and helpful assistant, like ChatGPT."},
+                {"role": "user", "content": user_input}
+            ],
             temperature=0.7,  # Slightly lower temperature for more controlled responses
             max_tokens=1024,
             top_p=1,
@@ -41,24 +44,15 @@ def get_completion(user_input, chat_history):
         for chunk in completion:
             response += chunk.choices[0].delta.content or ""
         
-        chat_history.append(("user", user_input))
-        chat_history.append(("assistant", response.strip()))  # Add assistant's response to history
-        return response.strip(), chat_history
-    
+        return response.strip()  # Clean up response
     except Exception as e:
         logger.error(f"Error during completion: {e}")
-        return "Sorry, I encountered an error while processing your request.", chat_history
+        return "Sorry, I encountered an error while processing your request."
 
 # Launch Gradio interface
 def launch_interface():
-    chat_history = []
-
-    def update_interface(user_input):
-        response, chat_history = get_completion(user_input, chat_history)
-        return response, chat_history
-
     demo = gr.Interface(
-        fn=update_interface,
+        fn=get_completion,
         inputs=gr.Textbox(
             label="Ask me anything:",
             placeholder="I am here to help! Ask away...",
@@ -67,19 +61,21 @@ def launch_interface():
             show_label=True,
             interactive=True
         ),
-        outputs=[
-            gr.Chatbot(label="Chat History").style(height=500),  # Display conversation history
-            gr.Textbox(label="Assistant's Response", interactive=False, show_label=True, lines=6, max_lines=10)
-        ],
+        outputs=gr.Textbox(
+            label="Response:",
+            interactive=False,
+            show_label=True,
+            lines=6,
+            max_lines=10
+        ),
         title="Chat with Mr AI",
         description="I am your friendly assistant, just like ChatGPT! Ask me anything, and I will do my best to help.",
-        theme="huggingface",  # Modern theme for the chat interface
+        theme="huggingface",  # More modern theme
         css="""
             .gr-box { border-radius: 15px; border: 1px solid #e1e1e1; padding: 20px; background-color: #f9f9f9; }
             .gr-button { background-color: #4CAF50; color: white; font-size: 14px; }
             .gr-textbox { border-radius: 8px; font-size: 16px; padding: 10px; }
             .gr-output { background-color: #f1f1f1; border-radius: 8px; font-size: 16px; padding: 15px; }
-            .gr-chatbot { background-color: #fafafa; padding: 20px; border-radius: 10px; }
         """,
         allow_flagging="never",
         live=True,  # Enable live updates if supported