Update app.py

app.py

@@ -10,46 +10,41 @@ if api_key is None:
 
 openai.api_key = api_key
 
-def get_python_help(history):
+def generate_exercise_question(topic):
     try:
-        # Construct the messages from the history
-        messages = [{"role": "system", "content": "You are an AI tutor specialized in teaching Python programming. Your primary task is to create exercise questions based on the topics users want to learn. Provide clear, concise, and informative exercises. If you are unsure about something, acknowledge it and suggest checking the official Python documentation for the most accurate information. Please ensure that all your responses strictly adhere to the subject of Python programming. Do not engage in any discussions that are insensitive, sexual, casual, comedic, or unrelated to Python programming. If a user asks a question outside of the Python programming scope, politely inform them that you are programmed to assist only with Python-related queries."}]
-        for h in history:
-            messages.append({"role": "user", "content": h[0]})
-            if h[1] is not None:
-                messages.append({"role": "assistant", "content": h[1]})
-        
-        messages.append({"role": "user", "content": history[-1][0]})
+        # Construct the initial message for exercise generation
+        messages = [{"role": "system", "content": "You are an AI tutor specialized in teaching Python programming. Your primary task is to create exercise questions based on the topics users want to learn. Provide clear, concise, and informative exercises. Ensure that all your responses strictly adhere to the subject of Python programming and do not engage in any discussions that are insensitive, sexual, casual, comedic, or unrelated to Python programming."}]
+        messages.append({"role": "user", "content": f"Create an exercise question on {topic}."})
 
         # Generate a response using gpt-3.5-turbo
         response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
             messages=messages,
-            max_tokens=500,
+            max_tokens=150,
             n=1,
             stop=None,
             temperature=0.7,
-            stream=True
+            stream=False
         )
 
-        # Stream the response
-        reply = ""
-        for chunk in response:
-            reply += chunk['choices'][0]['delta'].get('content', '')
-
-        history[-1] = (history[-1][0], reply)
-        return history
+        question = response.choices[0].message['content'].strip()
+        return question
     except Exception as e:
-        error_message = f"Error: {str(e)}"
-        history.append(("Error", error_message))
-        return history
+        return f"Error: {str(e)}"
 
 def respond(message, chat_history):
     if chat_history is None:
         chat_history = []
-    chat_history.append((message, None))
-    updated_history = get_python_help(chat_history)
-    return "", updated_history, updated_history
+    
+    # Check if this is the first message to set the topic
+    if len(chat_history) == 0:
+        topic = message
+        question = generate_exercise_question(topic)
+        chat_history.append((message, question))
+    else:
+        chat_history.append((message, "Please answer the question or type 'next' for another question on the same topic."))
+    
+    return "", chat_history, chat_history
 
 def undo(chat_history):
     if chat_history:
@@ -71,7 +66,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             user_input = gr.Textbox(
                 show_label=False,
-                placeholder="Type the topic you want to learn in Python and get exercise questions...",
+                placeholder="Type the Python topic you want to learn about...",
                 lines=1
             )
             submit_button = gr.Button("Submit")
@@ -83,11 +78,11 @@ with gr.Blocks() as demo:
         clear_button.click(clear, None, [chatbot, state])
 
     examples = [
-        ["Create exercise questions on Python lists"],
-        ["Give me an exercise on Python decorators"],
-        ["I need exercises on Python dictionaries"],
-        ["Generate questions on Python file handling"],
-        ["Create exercises on Python classes and objects"]
+        ["lists"],
+        ["decorators"],
+        ["dictionaries"],
+        ["file handling"],
+        ["classes and objects"]
     ]
 
     # Add examples