Fixed OpenAI API call and handled response

app.py

@@ -52,34 +52,39 @@ def handle_fda_query(query):
     logits = model(**inputs).logits
     return "FDA Query Processed: Contains regulatory info." if torch.argmax(logits, dim=1).item() == 1 else "FDA Query Processed: General."
 
-# Enhance via GPT-4o-mini
+# Function to enhance info via GPT-4o-mini
 def enhance_with_gpt4o(fda_response):
-    response = openai.ChatCompletion.create(
-        model="gpt-4o-mini",
-        messages=[{"role": "system", "content": "You are an expert FDA assistant."}, {"role": "user", "content": f"Enhance this FDA info: {fda_response}"}],
-        max_tokens=150
-    )
-    return response['choices'][0]['message']['content']
-
-# Initialize Hugging Face Inference Client
-client = InferenceClient("gpt-4o-mini")  # Correct model, make sure to replace with your actual model
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",  # Correct model
+            messages=[{"role": "system", "content": "You are an expert FDA assistant."}, {"role": "user", "content": f"Enhance this FDA info: {fda_response}"}],
+            max_tokens=150
+        )
+        return response['choices'][0]['message']['content']
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 def respond(message, system_message, max_tokens, temperature, top_p):
     try:
         # First retrieve info via PubMedBERT
         fda_response = handle_fda_query(message)
         
-        # Stream the enhanced response via GPT-4o-mini using the client
+        # Stream the enhanced response via GPT-4o-mini using the correct OpenAI API
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are an expert FDA assistant."},
+                {"role": "user", "content": f"Enhance this FDA info: {fda_response}"}
+            ],
+            max_tokens=max_tokens,
+            temperature=temperature,
+            top_p=top_p
+        )
+
         enhanced_response = ""
-<<<<<<< Updated upstream
         for chat_message in client.chat_completion(...):
             payload = json.loads(chat_message.lstrip("data:").rstrip("\n"))
             enhanced_response += payload["content"]  # Or however the payload structure works
-=======
-        for chat_message in client.chat_completion(...):  # Add params if needed
-            payload = clean_payload(chat_message)
-            enhanced_response += payload["content"]
->>>>>>> Stashed changes
 
         # Return both the PubMedBERT result and the enhanced version
         return f"Original Info from PubMedBERT: {fda_response}\n\nEnhanced Info via GPT-4o-mini: {enhanced_response}"