Update app.py

app.py

@@ -1,21 +1,15 @@
 import os
 import re
 import logging
-from fastapi import FastAPI, HTTPException, BackgroundTasks
-#from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import StreamingResponse
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import RedirectResponse
-import subprocess
 from pydantic import BaseModel
 from langchain.chains import RetrievalQA
 from langchain.prompts import PromptTemplate
 from langchain_community.llms import CTransformers
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
-import streamlit as st
-import uvicorn
-from threading import Thread
-import requests
+import subprocess
 from dotenv import load_dotenv
 
 # Load environment variables
@@ -25,11 +19,9 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-
 # FastAPI app
 app = FastAPI()
 
-
 # Load embeddings and vector database
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={"device": "cpu"})
 try:
@@ -82,25 +74,18 @@ class AnswerResponse(BaseModel):
 def clean_answer(answer):
     # Remove unnecessary characters and symbols
     cleaned_answer = re.sub(r'[^\w\s.,-]', '', answer)
-
     # Remove repetitive phrases by identifying repeated words or sequences
     cleaned_answer = re.sub(r'\b(\w+)( \1\b)+', r'\1', cleaned_answer)
-
     # Remove any trailing or leading spaces
     cleaned_answer = cleaned_answer.strip()
-
     # Replace multiple spaces with a single space
     cleaned_answer = re.sub(r'\s+', ' ', cleaned_answer)
-
     # Replace \n with newline character in markdown
     cleaned_answer = re.sub(r'\\n', '\n', cleaned_answer)
-
     # Check for bullet points and replace with markdown syntax
     cleaned_answer = re.sub(r'^\s*-\s+(.*)$', r'* \1', cleaned_answer, flags=re.MULTILINE)
-
     # Check for numbered lists and replace with markdown syntax
     cleaned_answer = re.sub(r'^\s*\d+\.\s+(.*)$', r'1. \1', cleaned_answer, flags=re.MULTILINE)
-
     # Check for headings and replace with markdown syntax
     cleaned_answer = re.sub(r'^\s*(#+)\s+(.*)$', r'\1 \2', cleaned_answer, flags=re.MULTILINE)
 
@@ -135,14 +120,12 @@ async def query(question_request: QuestionRequest):
         # Clean up the answer
         cleaned_answer = clean_answer(answer)
 
-        # Return cleaned_answer wrapped in a dictionary
         return {"answer": cleaned_answer}
 
     except Exception as e:
         logger.error(f"Error processing query: {e}")
         raise HTTPException(status_code=500, detail="Internal Server Error")
 
-
 def run_streamlit():
     subprocess.Popen(["streamlit", "run", "frontend.py", "--server.port", "8501"])
 
@@ -154,6 +137,5 @@ async def startup_event():
 async def root():
     return RedirectResponse(url="http://localhost:8501")
 
-
 #if __name__ == '__main__':
     #uvicorn.run(app, host='0.0.0.0', port=7860)