Update app.py

app.py

@@ -1,41 +1,59 @@
 import os
-import openai
+from openai import OpenAI
 import gradio as gr
 
 # Initialize OpenAI client with Groq model
 XAI_API_KEY = os.getenv("XAI_API_KEY")
-openai.api_key = XAI_API_KEY
+client = OpenAI(
+    api_key=XAI_API_KEY,
+    base_url="https://api.x.ai/v1",
+)
 
-# Define a function to handle chat with humor
-def chat_with_buddy(user_message):
+# Define a function to handle chat with humor, including chat history
+def chat_with_buddy(history, user_message):
+    # Start the conversation with a system message and add previous chat history
+    messages = [{"role": "system", "content": "You are Chat Buddy, a friendly and humorous chatbot who loves to make people smile."}]
+    messages += [{"role": "user", "content": msg[0]} for msg in history] + [{"role": "assistant", "content": msg[1]} for msg in history if len(msg) > 1]
+    messages.append({"role": "user", "content": user_message})
+    
     try:
-        # Generate a response from Groq with a humorous twist
-        response = openai.ChatCompletion.create(
+        # Generate a response from Groq
+        response = client.ChatCompletion.create(
             model="grok-beta",
-            messages=[
-                {"role": "system", "content": "You are Chat Buddy, a friendly and humorous chatbot who loves to make people smile."},
-                {"role": "user", "content": user_message},
-            ]
+            messages=messages
         )
-        # Extract and return the response content with a hint of humor
+        # Extract the response content
         reply = response.choices[0].message["content"]
-        
+
         # Add a humorous touch to the response
         if not reply.endswith("😄"):
-            reply += " 😄"  # Ensure a smile at the end of each response
-        return reply
+            reply += " 😄"
+
+        # Update chat history with the new user message and assistant response
+        history.append((user_message, reply))
+        return history
     
     except Exception as e:
-        return f"Oops, something went wrong! Just blame it on the robots... 😉 (Error: {str(e)})"
-
-# Set up the Gradio interface
-interface = gr.Interface(
-    fn=chat_with_buddy,          # Function to call
-    inputs="text",               # Input type
-    outputs="text",              # Output type
-    title="Chat Buddy",          # Title for the interface
-    description="Your friendly chatbot with a touch of humor! Ask Chat Buddy anything, and enjoy a chat that’s both helpful and light-hearted."
-)
+        # In case of error, add error message to history
+        error_message = f"Oops, something went wrong! Just blame it on the robots... 😉 (Error: {str(e)})"
+        history.append((user_message, error_message))
+        return history
+
+# Set up the Gradio chat interface with history
+with gr.Blocks() as chat_interface:
+    gr.Markdown("# Chat Buddy")
+    gr.Markdown("Your friendly chatbot with a touch of humor! Ask Chat Buddy anything, and enjoy a chat that’s both helpful and light-hearted.")
+    
+    chatbot = gr.Chatbot()
+    message = gr.Textbox(placeholder="Type your message here...", label="Your Message")
+    submit_button = gr.Button("Send")
+
+    # Define the submit action to update chat history
+    def submit(user_message, chat_history):
+        return chat_with_buddy(chat_history, user_message), ""
+
+    # Link the submit button to the function
+    submit_button.click(submit, inputs=[message, chatbot], outputs=[chatbot, message])
 
-# Launch the app
-interface.launch()
+# Launch the chat interface
+chat_interface.launch()