Update app.py

app.py

@@ -504,6 +504,8 @@ state = os.environ.get("state")
 system = os.environ.get("system")
 auth = os.environ.get("auth")
 auth2 = os.environ.get("auth2")
+openai.api_key = os.environ.get("auth")
+openai.api_base = os.environ.get("host")
 data = None
 
 np.set_printoptions(suppress=True)
@@ -598,13 +600,23 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                     }
 
                     try:
-                        response = requests.post(host, headers=header, json={
-                            "messages": messages,
-                            "model": model_llm
-                        }).json()
-                        print("RESPONSE TRY", response)
+                        # response = requests.post(host, headers=header, json={
+                        #     "messages": messages,
+                        #     "model": model_llm
+                        # }).json()
+                        
+                        completion = openai.ChatCompletion.create(
+                            model="gpt-3.5-turbo",
+                            messages=messages
+                        )
+        
+            
                         # reply = response["choices"][0]["message"]["content"]
-                        reply = response.choices[0].message['content']
+                        reply = completion.choices[0].message['content']
+                        # # reply = response["choices"][0]["message"]["content"]
+                        # reply = response.choices[0].message['content']
+                        print("RESPONSE TRY", completion)
+
 
                         output.append({"Mode": "Image", "type": max_label, "prediction_value": max_rounded_prediction, "content": reply})
                     except:
@@ -642,13 +654,19 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             }
 
             try:
-                response = requests.post(host, headers=headers, json={
-                    "messages": messages,
-                    "model": model_llm
-                }).json()
+                # response = requests.post(host, headers=headers, json={
+                #     "messages": messages,
+                #     "model": model_llm
+                # }).json()
+
+                completion = openai.ChatCompletion.create(
+                    model="gpt-3.5-turbo",
+                    messages=messages
+                )
+
     
                 # reply = response["choices"][0]["message"]["content"]
-                reply = response.choices[0].message['content']
+                reply = completion.choices[0].message['content']
             except:
                 reply = "Maximum messages: 15. Please clear your history and Try Again!"
             output.append({"Mode": "Chat", "content": reply})