Update app.py

app.py

@@ -5,18 +5,25 @@ import gradio as gr
 # Set your API key
 os.environ['XAI_API_KEY'] = 'xai-Gxw7oW4yR6Q6oTd0v9lDRLotXZQYJNz9YKlH7R6eMyTmqIV9h6uustEGZAaJEvGmewlwbUnM1jTX4chj'  # Replace with your actual API key
 
+# Initialize conversation history
+conversation_history = []
+
 def chat_with_bot(user_input):
+    # Append user input to conversation history
+    conversation_history.append({"role": "user", "content": user_input})
+
     url = "https://api.x.ai/v1/chat/completions"
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {os.environ['XAI_API_KEY']}"
     }
     
+    # Prepare messages for the API call
+    messages = [{"role": "system", "content": "You are a helpful assistant."}] + \
+               [{"role": msg["role"], "content": msg["content"]} for msg in conversation_history]
+
     data = {
-        "messages": [
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": user_input}
-        ],
+        "messages": messages,
         "model": "grok-beta",
         "stream": False,
         "temperature": 0.7
@@ -25,12 +32,26 @@ def chat_with_bot(user_input):
     response = requests.post(url, headers=headers, json=data)
     
     if response.status_code == 200:
-        return response.json()['choices'][0]['message']['content']
+        bot_response = response.json()['choices'][0]['message']['content']
+        # Append bot response to conversation history
+        conversation_history.append({"role": "assistant", "content": bot_response})
+        
+        # Format the chat history for display
+        chat_display = "\n".join([f"{msg['role'].capitalize()}: {msg['content']}" for msg in conversation_history])
+        
+        return chat_display
     else:
         return f"Error: {response.text}"
 
 # Create Gradio interface
-iface = gr.Interface(fn=chat_with_bot, inputs="text", outputs="text", title="xAI Chatbot", description="Chat with Grok!")
+iface = gr.Interface(
+    fn=chat_with_bot,
+    inputs="text",
+    outputs="text",
+    title="xAI Chatbot",
+    description="Chat with Grok! Type your message below.",
+    theme="default",
+)
 
 # Launch the interface
 if __name__ == "__main__":