Update app.py

app.py

@@ -3,18 +3,23 @@ from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 import uvicorn
 
-
+# Initialize FastAPI app
 app = FastAPI()
 
+# Define a function for the chat interface
+def respond_to_chat(history, message):
+    response = f"Hello, {message}!"
+    return response, history
 
-block = gr.Blocks().queue()
-with block:
-    prompt = gr.Text()
-    output = gr.Text()
-    
-    btn = gr.Button("Generate")
-    btn.click(lambda x:x, inputs=[prompt], outputs=[output])
+# Create Gradio ChatInterface
+chat = gr.ChatInterface(fn=respond_to_chat, title="Chat with AI")
 
+# Mount static files
 app.mount("/static", StaticFiles(directory="static", html=True), name="static")
-app = gr.mount_gradio_app(app, block, "/", gradio_api_url="http://localhost:7860/")
-uvicorn.run(app, host="0.0.0.0", port=7860)
+
+# Mount Gradio ChatInterface to FastAPI app
+app = gr.mount_gradio_app(app, chat, "/", gradio_api_url="http://localhost:7860/")
+
+# Run the app with uvicorn
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)