Update app.py

app.py

@@ -65,6 +65,36 @@ def generate_response(user_query, relevant_segment):
     """
     Generate a response emphasizing the bot's capability to provide information related to composting food.
     """
+    try:
+        # Define the messages
+        system_message = "You are a chatbot specialized in providing information about food composting tips, tricks, and basics."
+        user_message = f"Here's the information on composting: {relevant_segment} {user_query}"
+        messages = [
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": user_message}
+        ]
+        
+        # Generate the response
+        response = openai.ChatCompletion.create(
+            model="text-davinci-003",
+            messages=messages,
+            max_tokens=150,
+            temperature=0.2,
+            top_p=1
+        )
+        
+        # Extract the generated text and confidence score
+        generated_text = response['choices'][0]['message']['content'].strip()
+        # Confidence score extraction may vary depending on the model and API response structure
+        confidence_score = response['choices'][0].get('logprobs', {}).get('token_logprobs', [None])[0]
+
+        return generated_text, confidence_score
+    
+    except Exception as e:
+        print(f"Error in generating response: {e}")
+        return f"Error in generating response: {e}"
+
+    '''
     try:
         system_message = "You are a chatbot specialized in providing information about food composting tips, tricks, and basics."
         user_message = f"Here's the information on composting: {relevant_segment}"
@@ -85,6 +115,7 @@ def generate_response(user_query, relevant_segment):
     except Exception as e:
         print(f"Error in generating response: {e}")
         return f"Error in generating response: {e}"
+    '''
 
 
 def query_model(question):
@@ -119,7 +150,7 @@ topics = """
 
 # Define the HTML iframe content
 iframe = '''
-    <div style="height:20px;"></div>
+    <div style="height:15px;"></div>
     <iframe style="border-radius:12px" 
     src="https://open.spotify.com/embed/episode/1Emjgqf8PfwD42kvyKvtfW?utm_source=generator&theme=0" 
     width="100%" height="152" frameBorder="0"