Update main.py

main.py

@@ -91,7 +91,32 @@ We ensure to keep you updated at each stage and incorporate your feedback to del
 use this details to give answer for my questio.only give system response only(not include customer message)
 question : """
 
+import google.generativeai as genai
+from langchain.llms import OpenAI
+from langchain.chat_models import ChatOpenAI
+from langchain.agents.agent_types import AgentType
+#from langchain_experimental.agents.agent_toolkits import create_csv_agent
+
+
+from llama_index.llms import OpenAI
+from llama_index import VectorStoreIndex, SimpleDirectoryReader
+from llama_index.llms import OpenAI
+from llama_index import StorageContext, load_index_from_storage
+
 
+os.environ["OPENAI_API_KEY"]
+
+try:
+    storage_context = StorageContext.from_defaults(persist_dir="llama_index")
+    index = load_index_from_storage(storage_context=storage_context)
+    print("loaded")
+except:     
+    documents = SimpleDirectoryReader("userguid").load_data()
+    index = VectorStoreIndex.from_documents(documents)
+    index.storage_context.persist("llama_index")
+    print("index created")
+
+query_engine = index.as_query_engine()
 
 app = FastAPI()
 
@@ -109,14 +134,46 @@ async def reply_whatsapp(request: Request):
     num_media = int(form_data.get("NumMedia", 0))
     from_number = form_data.get("From")
     message_body = form_data.get("Body")
+    user_query = message_body
 
     gen_response = model.generate_content(str(prompt)+message_body)
 
     response = MessagingResponse()
 
-    msg = response.message(gen_response.text)
+    
     #msg.media(GOOD_BOY_URL)
-
-    return PlainTextResponse(str(response), media_type="application/xml")
+    gpt_response = query_engine.query("""        
+          if you find the answer from provided data then give answer with steps and make the more details link within the <a href>lank hyper link.
+          if not find the answer from provided data then say 'please contact our helpdesk' \n\n
+          user question : """+user_query)
+
+    print(str(gpt_response).lower())
+    if "please contact our helpdesk" in str(gpt_response).lower() or "please contact" in str(gpt_response).lower():
+            print("help desk option")
+
+            openai.api_key = os.environ["OPENAI_API_KEY"]
+
+            default = """<br><br>Dear<br>If you have a specific question or need assistance, please feel free to submit a ticket, and our support team will be happy to help you:<br><br>Submit a Ticket:<br>Email: support@storemate.lk<br>Hotline: 0114 226 999<br><br>Thank You """
+            messages = [{"role": "user", "content": user_query+".   always give small answers"}]
+            gpt_response = openai.chat.completions.create(
+            
+            model="gpt-3.5-turbo",
+            
+            messages=messages,
+            
+            temperature=0,
+            
+            )
+            
+            msg = response.message(str(gpt_response.choices[0].message.content) + default)
+            return PlainTextResponse(str(response), media_type="application/xml")
+            
+    result = ""
+    for lines in str(gpt_response).split("\n"):
+            result = result +"<p>"+lines+"</p><br>"
+    msg = response.message(result)   
+    return PlainTextResponse(str(response), media_type="application/xml")   
+
+    
 
 # Run the application (Make sure you have the necessary setup to run FastAPI)