Update main.py

main.py

@@ -8,88 +8,7 @@ import google.generativeai as genai
 secret = os.environ["key"]
 
 genai.configure(api_key=secret)
-model = genai.GenerativeModel('gemini-pro')
-
-GOOD_BOY_URL = (
-    "https://images.unsplash.com/photo-1518717758536-85ae29035b6d?ixlib=rb-1.2.1"
-    "&ixid=eyJhcHBfaWQiOjEyMDd9&auto=format&fit=crop&w=1350&q=80"
-)
-
-prompt = """Certainly! Below is a chat conversation template for your AI company, Nuron, specialized in developing AI and ML solutions and chatbot systems. The conversation includes typical customer inquiries and responses that can be used during business hours (9 AM to 5 PM, Monday to Friday).
-
----
-
-**Customer**: Hi, I'm interested in learning more about your AI and ML solutions. Can you tell me more?
-
-**Nuron Support**: Hello! Thank you for reaching out to Nuron. We specialize in developing cutting-edge AI and ML solutions tailored to meet your business needs. Our services include predictive analytics, natural language processing, computer vision, and custom chatbot systems. How can we assist you today?
-
----
-
-**Customer**: Who is behind Nuron?
-
-**Nuron Support**: Nuron was founded by Arafath, our CEO, who has extensive experience in AI and ML technologies. Under his leadership, we have delivered numerous successful projects across various industries.
-
----
-
-**Customer**: What are your working hours?
-
-**Nuron Support**: Our working hours are from 9 AM to 5 PM, Monday to Friday. During these hours, our team is available to assist you with any inquiries or support you might need.
-
----
-
-**Customer**: Can you develop a custom chatbot for my business?
-
-**Nuron Support**: Absolutely! We specialize in creating custom chatbot systems designed to enhance customer engagement and streamline business processes. Please provide us with some details about your requirements, and we'll be happy to discuss how we can help.
-
----
-
-**Customer**: How can I get in touch with Arafath?
-
-**Nuron Support**: You can reach out to Arafath by emailing our support team at [email protected], and we will ensure your message is forwarded to him. Alternatively, you can schedule a meeting through our website.
-
----
-
-**Customer**: What industries do you serve?
-
-**Nuron Support**: We have experience working with a diverse range of industries including healthcare, finance, retail, and manufacturing. Our AI and ML solutions are adaptable and can be customized to fit the unique needs of any sector.
-
----
-
-**Customer**: How do I get started with a project at Nuron?
-
-**Nuron Support**: To get started, simply contact us through our website or email us at [email protected] with a brief description of your project. We'll arrange a consultation to understand your requirements and provide you with a tailored proposal.
-
----
-
-**Customer**: What is the process for developing an AI solution with Nuron?
-
-**Nuron Support**: Our process typically involves the following steps:
-1. Initial consultation to understand your needs.
-2. Proposal and project planning.
-3. Data collection and analysis.
-4. Model development and training.
-5. Implementation and integration.
-6. Testing and validation.
-7. Deployment and support.
-
-We ensure to keep you updated at each stage and incorporate your feedback to deliver the best solution.
-
----
-
-**Customer**: Can you provide support after the project is completed?
-
-**Nuron Support**: Yes, we offer comprehensive post-deployment support and maintenance services to ensure that your AI and ML solutions continue to perform optimally. Our team is always here to assist you with any updates or issues you may encounter.
-
----
-
-**Customer**: Thank you for the information!
-
-**Nuron Support**: You're welcome! If you have any more questions or need further assistance, feel free to contact us. Have a great day!
-
----
-
-use this details to give answer for my questio.only give system response only(not include customer message)
-question : """
+model = genai.GenerativeModel('gemini-1.5-pro')
 
 import user_guide_sync
 
@@ -112,6 +31,7 @@ async def update_user_guide_data():
     global query_engine,index
     index = user_guide_sync.update_user_guide()
     query_engine = index.as_query_engine()
+    return "guide updated"
 
 @app.post("/whatsapp")
 async def reply_whatsapp(request: Request):
@@ -121,52 +41,62 @@ async def reply_whatsapp(request: Request):
     message_body = form_data.get("Body")
     user_query = message_body
 
-    gen_response = model.generate_content(str(prompt)+message_body)
-
     response = MessagingResponse()
-
-    
     #msg.media(GOOD_BOY_URL)
-    gpt_response = query_engine.query(f"""        
-          if you find the answer from provided data then give the realistic(like real human) answer with steps and add the more details link and propper line breaks(\n).
-          if not find the answer from provided data then say 'please contact our helpdesk'
-          
-          user question : {user_query}""")
-    
-    default = """Dear\n\nIf 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 \n\nSubmit a Ticket: \n\tEmail: [email protected]\n\tHotline: 0114 226 999\nThank You """
-
-    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"]
-
-            messages = [
-                {"role": "system", "content": "you are parallax technologies chatbot design for answer the user question like a real human"},
-                {"role": "user", "content": user_query+".   always give small and realistic response"}
-            ]
+    try:
+        gpt_response = query_engine.query(f"""        
+              if you find the answer from provided data then give the realistic(like real human) answer with steps and add the more details link and propper line breaks(\n).
+              if not find the answer from provided data then say 'please contact our helpdesk'
+              
+              user question : {user_query}""")
         
-            # gpt_response = openai.chat.completions.create(
-            
-            # model="gpt-3.5-turbo",
-            
-            # messages=messages,
-            
-            # temperature=0,
-            
-            # )
-            
-            # response.message(str(gpt_response.choices[0].message.content))
-            response.message(str(default))
-            return PlainTextResponse(str(response), media_type="application/xml")
+        default = """Dear\n\nIf 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 \n\nSubmit a Ticket: \n\tEmail: support@storemate.lk\n\tHotline: 0114 226 999\nThank You """
+    
+        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"]
+    
+                messages = [
+                    {"role": "system", "content": "you are parallax technologies chatbot design for answer the user question like a real human"},
+                    {"role": "user", "content": user_query+".   always give small and realistic response"}
+                ]
+                prompt = f"""
+                         system:
+                         you are parallax technologies chatbot design for answer the user question like a real human for only greedings.
+                         if user ask other questions say "i don't know"
+                         only give single short & small answer
+                         
+                         user:{user_query}  
+                         """
+                gen_response = model.generate_content(prompt)
             
-    #result = ""
-    #for lines in str(gpt_response).split("\n"):
-            #result = result +"<p>"+lines+"</p><br>"
-    response.message(str(gpt_response)) 
-    #response.message("https://storemate.lk")
-    return PlainTextResponse(str(response), media_type="application/xml")   
-
+                # gpt_response = openai.chat.completions.create(
+                
+                # model="gpt-3.5-turbo",
+                
+                # messages=messages,
+                
+                # temperature=0,
+                
+                # )
+                
+                # response.message(str(gpt_response.choices[0].message.content))
+                response.message(gen_response.text)
+                response.message(str(default))
+                return PlainTextResponse(str(response), media_type="application/xml")
+                
+        #result = ""
+        #for lines in str(gpt_response).split("\n"):
+                #result = result +"<p>"+lines+"</p><br>"
+        response.message(str(gpt_response)) 
+        #response.message("https://storemate.lk")
+        return PlainTextResponse(str(response), media_type="application/xml")   
+    except Exception as e:
+        print(str(e))
+        response.message("please ask again...!")
+        return PlainTextResponse(str(response), media_type="application/xml")  
     
 
 # Run the application (Make sure you have the necessary setup to run FastAPI)