Update app.py

app.py

@@ -1,20 +1,27 @@
 import gradio as gr
 from groq import Groq
 import os
+import time
 
 api_key = os.getenv('GROQ_API_KEY')
 # Initialize Groq client
 client = Groq(api_key=api_key)
 
 # Function to generate responses with error handling
-def generate_response(user_input, chat_history):
+def generate_response(user_input, chat_history: list):
     try:
         # Prepare messages with chat history
-        messages = [{"role": "system", "content": "You are a helpful mental health assistant."}]
-        for user_message, bot_response in chat_history:
-            messages.append({"role": "user", "content": user_message})
-            messages.append({"role": "assistant", "content": bot_response})
-        messages.append({"role": "user", "content": user_input})
+        messages = [{"role": "system", "content": "You are a mental health assistant. Your responses should be empathetic, non-judgmental, and provide helpful advice based on mental health principles. Always encourage seeking professional help when needed."}]
+        
+        # Iterate through chat history and add user and assistant messages
+        for message in chat_history:
+            # Ensure that each message contains only 'role' and 'content' keys
+            if 'role' in message and 'content' in message:
+                messages.append({"role": message["role"], "content": message["content"]})
+            else:
+                print(f"Skipping invalid message: {message}")
+        
+        messages.append({"role": "user", "content": user_input})  # Add the current user message
 
         # Call Groq API to get a response from LLaMA
         chat_completion = client.chat.completions.create(
@@ -24,7 +31,7 @@ def generate_response(user_input, chat_history):
         
         # Extract response
         response = chat_completion.choices[0].message.content
-        return response, chat_history
+        return response, chat_history  # Ensure you return both response and chat_history
 
     except Exception as e:
         print(f"Error occurred: {e}")  # Print error to console for debugging
@@ -33,35 +40,42 @@ def generate_response(user_input, chat_history):
 # Define Gradio interface
 def gradio_interface():
     with gr.Blocks() as demo:
-        # Variable to store chat history
+        # Initialize chat history
         chat_history = []
 
-        # Modify `generate_and_clear` to accept additional unused arguments
-        def generate_and_clear(user_input, *args, **kwargs):
-            response, updated_history = generate_response(user_input, chat_history)
-            chat_history.append((user_input, response))
-            return response
+        # Create input textbox and button for clearing chat
+        gr.Markdown("## Mental Health Chatbot")
+        chatbot = gr.Chatbot(type="messages")
+        msg = gr.Textbox(placeholder="Type your message here...")
+        clear = gr.Button("Clear")
 
-        def clear_chat():
-            chat_history.clear()
-            return "Chat history cleared. Start a new conversation!"
+        # User message submission function
+        def user(user_message, history: list):
+            # Add user message to the history
+            history.append({"role": "user", "content": user_message})
+            return "", history  # Reset message input and return updated history
 
-        # Title and instructions
-        gr.Markdown("## Mental Health Chatbot - Powered by LLaMA on Groq")
+        # Bot response function with simulated typing effect
+        def bot(history: list):
+            # Ensure that history is not empty
+            if len(history) > 0:
+                user_input = history[-1]["content"]  # Get the last user message
+                response, updated_history = generate_response(user_input, history)  # Get bot's response
 
-        # Chat interface setup
-        chatbot = gr.ChatInterface(
-            fn=generate_and_clear,
-            #additional_inputs=[gr.Textbox(placeholder="Enter your message here...", label="Your Message")],
-            #title="Mental Health Chatbot"
-        )
+                history = updated_history  # Update the history with the new response
+                history.append({"role": "assistant", "content": ""})  # Add placeholder for assistant
 
-        # Add button to clear chat history
-        clear_button = gr.Button("Clear Chat History")
-        clear_button.click(fn=clear_chat, inputs=[], outputs=chatbot)  # Clear chat on button click
+                # Simulate typing effect for the bot's response
+                for character in response:
+                    history[-1]['content'] += character
+                    time.sleep(0.02)  # Typing delay
+                    yield history  # Yield updated history as the bot types
+
+        # Set up interaction flow:
+        msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(bot, chatbot, chatbot)
+        clear.click(lambda: [], None, chatbot, queue=False)  # Clear chat history when clicked
 
     demo.launch()
 
 # Run the interface
-gradio_interface()
-
+gradio_interface()