Update app.py

app.py

@@ -1,8 +1,4 @@
-# app.py
-
 import pandas as pd
-from fastapi import FastAPI
-from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
 import faiss
 from datasets import load_dataset
@@ -37,55 +33,17 @@ embeddings = model.encode(questions, convert_to_tensor=True)
 index = faiss.IndexFlatL2(embeddings.shape[1])
 index.add(embeddings.cpu().numpy())
 
-# Define FastAPI app and model
-app = FastAPI()
-
-class Query(BaseModel):
-    question: str
-
-@app.post("/ask")
-def ask_bot(query: Query):
-    question_embedding = model.encode([query.question], convert_to_tensor=True)
+def get_answer(question):
+    question_embedding = model.encode([question], convert_to_tensor=True)
     question_embedding_np = question_embedding.cpu().numpy()
     _, closest_index = index.search(question_embedding_np, k=1)
     best_match_idx = closest_index[0][0]
     answer = combined_data.iloc[best_match_idx]['Answer']
-    return {"answer": answer}
-
-# Gradio Interface
-
-import gradio as gr
-import requests
-
-# Define the URL of your FastAPI endpoint
-API_URL = "http://localhost:8000/ask"  # Update to your deployed FastAPI URL if needed
-
-def respond(message, history: list[tuple[str, str]]):
-    payload = {"question": message}
-    
-    try:
-        response = requests.post(API_URL, json=payload)
-        response.raise_for_status()
-        response_data = response.json()
-        answer = response_data.get("answer", "Sorry, I didn't get that.")
-    except requests.exceptions.RequestException as e:
-        answer = f"Request Error: {str(e)}"
-    
-    # Update history
-    history.append((message, answer))
-    return answer, history
-
-# Gradio Chat Interface
-demo = gr.ChatInterface(
-    respond,
-)
+    return answer if answer else "Answer not found."
 
+# Example usage
 if __name__ == "__main__":
-    import threading
-    import uvicorn
-
-    # Run FastAPI in a separate thread
-    threading.Thread(target=uvicorn.run, args=(app,), kwargs={"host": "0.0.0.0", "port": 8000}).start()
-    
-    # Launch Gradio interface
-    demo.launch()
+    # Replace this with any input method you want, like a chatbot interface
+    user_question = input("Ask your question: ")
+    response = get_answer(user_question)
+    print("Response:", response)