Update app.py

app.py

@@ -5,12 +5,12 @@ import requests
 # Function to call ChatGPT (OpenAI)
 def query_chatgpt(prompt):
     openai.api_key = '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'  # Add your OpenAI API Key
-    response = openai.Completion.create(
+    response = openai.chat.Completion.create(
         model="gpt-3.5-turbo",  # Updated model
-        prompt=prompt,
+        messages=[{"role": "user", "content": prompt}],
         max_tokens=150
     )
-    return response.choices[0].text.strip()
+    return response['choices'][0]['message']['content'].strip()
 
 # Function to query Gemini API (Google's Generative Language API)
 def query_gemini(prompt):
@@ -30,12 +30,16 @@ def query_gemini(prompt):
     
     # Check if request is successful
     if response.status_code == 200:
-        return response.json().get('content', 'No content returned')
+        content = response.json().get('content', 'No content returned')
+        if content:
+            return content
+        else:
+            return "No content returned from Gemini."
     else:
         return f"Error: {response.status_code}, {response.text}"
 
 # Streamlit layout
-st.title("Compare AI Models: ChatGPT and Gemini")
+st.title("ChatGPT and Gemini")
 prompt = st.text_area("Enter your query:")
 
 if st.button("ChatGPT"):