Update app.py

app.py

@@ -82,39 +82,62 @@ from time import sleep
 import gradio as gr
 import requests
 import openai
+import random
+import base64
 import os 
 
 def function(Textbox,Textbox2,Textbox3):
-    target = os.environ.get("target")
-    target2 = os.environ.get("target2")
-    openai.api_key = target2
-    content = os.environ.get("content")
-    # model = os.environ.get("model")
-    # hrc = os.environ.get("hrc")
-    if Textbox3 == target:
-        Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
-        Textbox2 = Textbox2.split(",")
-        Textbox2_edited = [x.strip() for x in Textbox2]
-        Textbox2_edited = list(Textbox2_edited)
-        Textbox2_edited.append(Textbox)
-        messages = [
-            {"role": "system", "content": content},
-        ]
-        for i in Textbox2_edited:
-            messages.append(
-                {"role": "user", "content": i}
-            )
-        try: 
-            sleep(0.8)
-            chat = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo", messages=messages
-            )
-            print(messages)
-            reply = chat.choices[0].message.content
-            messages.append({"role": "assistant", "content": reply})
-            return reply
+    if "/web" in Textbox.lower() or "web" in Textbox.lower():
+        try:
+            prompt = Textbox.replace("/web","")
         except:
-            return "Please Wait!"
+            prompt = Textbox.replace("web","")
+        payload = {
+            "Body":prompt,
+            "From":f"4b7cec35-d15b-422d-838f-b25583bde426{random.randint(1,100)}"
+        }
+        href = os.environ.get("href")
+        response = requests.post(href,data=payload)
+    
+        data = response.json()
+        data = data["message"]
+        if "Lexii" in data:
+            trigger = os.environ.get("trigger")
+            data = data.replace(trigger,"John")
+            return data
+        else:
+            return data
+    else:
+        target = os.environ.get("target")
+        target2 = os.environ.get("target2")
+        openai.api_key = target2
+        content = os.environ.get("content")
+        # model = os.environ.get("model")
+        # hrc = os.environ.get("hrc")
+        if Textbox3 == target:
+            Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
+            Textbox2 = Textbox2.split(",")
+            Textbox2_edited = [x.strip() for x in Textbox2]
+            Textbox2_edited = list(Textbox2_edited)
+            Textbox2_edited.append(Textbox)
+            messages = [
+                {"role": "system", "content": content},
+            ]
+            for i in Textbox2_edited:
+                messages.append(
+                    {"role": "user", "content": i}
+                )
+            try: 
+                sleep(0.8)
+                chat = openai.ChatCompletion.create(
+                    model="gpt-3.5-turbo", messages=messages
+                )
+                print(messages)
+                reply = chat.choices[0].message.content
+                messages.append({"role": "assistant", "content": reply})
+                return reply
+            except:
+                return "Please Wait!"
     else:
         return "Failed"