Update main.py

main.py

@@ -4,12 +4,95 @@ from fastapi.responses import StreamingResponse
 import requests
 import json,os,openai
 
+
 api_key = os.environ["OPENAI_API_KEY"]
 
 app = FastAPI()
 
+
 # Declare the continuous function as an async function.
-async def get_response():
-    return "hu"
+async def streaming(userText,idf):
+
+    if str(idf)=="1":
+      openaikey = "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"
+      openai.api_key = openaikey
+      print(idf)
+
+
+    if str(idf) == "2" and str(userText.split("Prospect:")[-1]).lower() == " no":
+        remaind_msg = "That's alright. We provide AI engineering as a service. We work with some top startups and Fortune 500 companies to develop custom AI models for their business. Does that ring a bell now?"
+        return remaind_msg
+        
+    if str(idf) == "2" and (
+            "remind" in str(userText.split("Prospect:")[-1]).lower() or
+            "remember" in str(userText.split("Prospect:")[-1]).lower() or
+            "forgot" in str(userText.split("Prospect:")[-1]).lower()
+        ):
+        remaind_msg = "That's alright. We provide AI engineering as a service. We work with some top startups and Fortune 500 companies to develop custom AI models for their business. Does that ring a bell now?"
+
+        return remaind_msg
+    
+    dec = ""
+    yest = "This is [SDR’s name] with Neural Leap. I saw you schedule a call with us for [Insert day and time] to learn more about our A.I. engineering services. Does that ring a bell?"
+
+    if str(idf) == "1":
+        #openai.api_key = openaikey
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {
+                    "role": "user",
+                    "content": "this is my part of conversation >" + userText +
+                               "\n\ntell me this Prospect like to continue the conversation or unlike to continue the conversation? give only like or unlike as a result"
+                }
+            ],
+        )
+
+        dec = str(response.choices[0].message.content)
+        if 'unlike' not in dec.lower():
+            return yest
+
+
+
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        stream=True,
+        messages=[{"role":"system","content":"""
+    \nif prospect like to finish or leave the conversation or call then lets finish don't ask any questions.give suitable respones
+
+    \n\n if prospect confirm the schedule the call then don't ask any questions only give this response one time only don't ask same questions again and again
+    "That's great! I just wanted to confirm that  (time) still works for you and you are able to be on. Is that still the case ? ".
+
+    \n\n if prospect want to cancel the time not reschedule or change time then only give this response
+    "I appreciate you letting me know. If this is about finding a suitable time, let's figure out a slot that works for both of us. I'm available for both morning and afternoon sessions this week, what do you you prefer?"
+
+    \n\nif prospect like to reschedule or change the meeting without specific time(am or pm) then always suggest 3 diffrents random time between 10 am to 5pm (example 10am , 12pm ,3pm like this) and don't include 'appologise' word in response.
+
+    \n\nif prospect like to reschedule or change the meeting with specific day and time(am or pm) then give like this response 'Understood. I can make (prospect provided time). I will update our schedule accordingly. Thank you for letting us know.'
+
+    \n\nIf the prospect confirms that the time still works for him, then ask the following question:
+    sample question format 'Before we begin our conversation, it would be helpful to know more about your financial landscape. Are there specific budgetary constraints or financial aspects that we should take into account to ensure our discussion is tailored to your needs?'
+
+    \n\nAfter the prospect provides budget details then only, inquire about and assess their specific needs and pain points.'
+
+    \n\nnote : don't ask same questions again and again always give single response, don't iclude the prospect, sdr conversation responses\n\n """},
+
+    {"role":"user","content":userText}],
+    )
+
+    for chunk in response:
+      try:
+        yield chunk.choices[0].delta.content
+      except:
+        yield "error"
+    
+
+
+
+
+@app.post("/get_sdr_response/")
+async def get_sdr_response(userText,idf):  
+    return StreamingResponse(streaming(userText,idf), media_type="text/plain")
+