Update frontend.py

frontend.py

@@ -2,14 +2,13 @@
 import requests
 import gradio as gr
 
-# Define the API URL
-API_URL = "http://localhost:8000/chat/"  # URL for your FastAPI backend
+# Define the API URL for FastAPI
+API_URL = "http://localhost:8000/chat/"  # Ensure this matches your FastAPI server
 
 def get_response(user_message):
     try:
         response = requests.post(API_URL, json={"message": user_message})
-        response_data = response.json()
-        return response_data.get("response", "Error: Unable to get response.")
+        return response.json().get("response", "Error: Unable to get response.")
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -18,8 +17,8 @@ iface = gr.Interface(
     fn=get_response,  # Function to call for generating responses
     inputs=gr.Textbox(label="Your Question", placeholder="Enter your message here..."),
     outputs=gr.Textbox(label="Chatbot Response"),
-    title="Chatbot with Hugging Face Transformers",
-    description="Chat with an AI assistant that uses Hugging Face Transformers to generate responses."
+    title="Chatbot with LLaMA API",
+    description="Chat with an AI assistant that uses the LLaMA model to generate responses."
 )
 
 # Launch the Gradio interface