Update app.py

app.py

@@ -44,6 +44,22 @@ def ask_rail_defect_question(question, model_name='ft:gpt-3.5-turbo-0125:persona
     )
     return response.choices[0].message['content']
 
+def ask_feedback(question, model_name='gpt-3.5-turbo'):
+    response = openai.ChatCompletion.create(
+        model=model_name,
+        messages=[
+            {
+                "role": "system",
+                "content": "The assistant is knowledgeable about user feedback and can analysis feedback and answer user-firndly way that they are updating the system message and LLM continuosly.",
+            },
+            {
+                "role": "user",
+                "content": question,
+            }
+        ],
+    )
+    return response.choices[0].message['content']
+
 def generate_variations(base_prompt, number_of_variations):
     locations = ["on the left side", "on the right side", "at the top", "at the bottom", "in the center"]
     sizes = ["small", "medium", "large", "tiny", "huge"]
@@ -106,17 +122,21 @@ with gr.Blocks() as app:
     feedback_input = gr.Textbox(label="Enter your feedback", placeholder="Write your feedback here...")
     like_button = gr.Button(value="👍 Like")
     dislike_button = gr.Button(value="👎 Dislike")
-    feedback_result = gr.Textbox(label="System Response", interactive=False)
+    feedback_result = gr.Textbox(label="System Response")
+    send_feedback_button = gr.Button("Send Feedback 🚀")
     refresh_button = gr.Button("Refresh Page")
     
 
     def handle_feedback(feedback, sentiment):
-        analysis = ask_rail_defect_question(feedback)
+        analysis = ask_feedback(feedback)
         if sentiment == "👍 Like":
             return f"Thank you for your positive feedback! Your feedback: '{feedback}' was analyzed as: '{analysis}'. We are updating our system based on your feedback."
-        else:
+        elif sentiment == "👎 Dislike":
             return f"We're sorry to hear that. Your feedback: '{feedback}' was analyzed as: '{analysis}'. We are updating our system based on your feedback."
-
+        else:
+            return f"Your feedback: '{feedback}' was analyzed as: '{analysis}'. We are updating our system based on your feedback."
+            
+    send_feedback_button.click(lambda x: handle_feedback(x, ""), inputs=feedback_input, outputs=feedback_result)
     like_button.click(lambda x: handle_feedback(x, "👍 Like"), inputs=feedback_input, outputs=feedback_result)
     dislike_button.click(lambda x: handle_feedback(x, "👎 Dislike"), inputs=feedback_input, outputs=feedback_result)
     refresh_button.click(lambda: gr.update(reload_browser=True))