Spaces:
Build error
Build error
# frontend.py | |
import requests | |
import gradio as gr | |
# 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}) | |
return response.json().get("response", "Error: Unable to get response.") | |
except Exception as e: | |
return f"Error: {str(e)}" | |
# Create a Gradio interface | |
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 LLaMA API", | |
description="Chat with an AI assistant that uses the LLaMA model to generate responses." | |
) | |
# Launch the Gradio interface | |
iface.launch() |