Update chatbot.py

chatbot.py

@@ -1,19 +1,28 @@
-# chatbot.py
-import requests
+# app.py
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from chatbot import Chatbot
 
-class Chatbot:
-    def __init__(self, api_url: str):
-        self.api_url = api_url
+# Initialize FastAPI
+app = FastAPI()
 
-    def generate_response(self, user_message: str) -> str:
-        try:
-            response = requests.post(self.api_url, json={"prompt": user_message})
-            return response.json().get("response", "Error: Unable to get response.")
-        except Exception as e:
-            return f"Error: {str(e)}"
+# Initialize the chatbot connected to the LLaMA API
+chatbot = Chatbot(api_url="http://localhost:8001/llama/")  # Adjust this if you change the port of api.py
 
-    def auto_generate_response(self):
-        # This is where you can set up any logic or triggers for automatic responses
-        # For example, using a fixed prompt
-        default_prompt = "Hello, how can I assist you today?"
-        return self.generate_response(default_prompt)
+class ChatRequest(BaseModel):
+    message: str
+
+@app.post("/chat/")
+async def chat(request: ChatRequest):
+    """
+    This endpoint could be called from another service if needed.
+    Automatically generates a response.
+    """
+    try:
+        # Automatically generate a response based on a default message or logic
+        bot_response = chatbot.auto_generate_response()
+        return {"response": bot_response}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# To run the backend, use 'uvicorn app:app --reload' in the terminal.