Update app.py

app.py

@@ -8,30 +8,26 @@ openai.api_key = os.environ.get("openai_api_key")
 
 # Define a function to generate responses using GPT-3.5 Turbo
 def generate_response(user_prompt):
-    # response = openai.ChatCompletion.create(
-    # engine="gpt432k",  #engine = "leadership",
-    # messages=[
-    #     {"role": "system", "content": "You are a helpful assistant."},
-       
-    #       {"role": "user", "content": prompt}
-    # ],
-    # temperature=0,
-    #     max_tokens=200
-    #     # stream = True
-        
-    # )
-    # text = response['choices'][0]['message']['content']
+
+    # Define the system message
+    system_msg = 'You are a helpful assistant.'
+
+# Define the user message
+     prompt= f'''I will give you a question and you detect which category does this question belong to. It should be from these categories - 
+        physical activity, sleep, nutrition and preventive care. Make sure you just reply with response in json format "category":"[sleep,nutrition]". 
+        Note that single question may belong to multiple categories. Dont add any opening lines just reply with json response. 
+        Question: {user_prompt}'''
+    #user_msg = 'Create a small dataset about total sales over the last year. The format of the dataset should be a data frame with 12 rows and 2 columns. The columns should be called "month" and "total_sales_usd". The "month" column should contain the shortened forms of month names from "Jan" to "Dec". The "total_sales_usd" column should contain random numeric values taken from a normal distribution with mean 100000 and standard deviation 5000. Provide Python code to generate the dataset, then provide the output in the format of a markdown table.'
+
+# Create a dataset using GPT
    
-    # return text
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",  # Use GPT-3.5 Turbo engine
-        prompt= f'''I will give you a question and you detect which category does this question belong to. It should be from these categories - 
-        physical activity, sleep, nutrition and preventive care. Make sure you just reply with response in json format "category":"[sleep,nutrition]". 
-        Note that single question may belong to multiple categories. Dont add any opening lines just reply with json response. 
-        Question: {user_prompt}''',
+       ,messages=[{"role": "system", "content": system_msg},
+                                         {"role": "user", "content": prompt}]
         max_tokens=100,  # You can adjust this to limit the response length
     )
-    return response['choices'][0]['message']['content']
+    return response["choices"][0]["message"]["content"]
 
 # Create a Gradio interface
 iface = gr.Interface(