Generate Button Behavior

app.py

@@ -2,59 +2,64 @@ import gradio as gr
 import time
 from transformers import pipeline
 
-# Load text generation pipeline
+# Load question generation model
 generator = pipeline("text-generation", model="gpt2")
 
-# Paradata tracking
+# Track paradata
 paradata = {}
 
-# Generate 5 survey questions from prompt
+# Generate 5 questions
 def generate_questions(prompt):
     input_text = f"Generate 5 survey questions on: {prompt}"
     result = generator(input_text, max_length=150, num_return_sequences=1)[0]['generated_text']
-    questions = [line.strip() for line in result.split('\n') if line.strip()]
-    return questions[:5]
+    lines = [line.strip() for line in result.split('\n') if line.strip()]
+    return lines[:5] + [""] * (5 - len(lines))  # Ensure 5 questions
 
-# Start timer for a question
+# Start timer
 def start_timer(qid):
     paradata[qid] = {"start_time": time.time()}
-    return f"Timer started for Question {qid+1}"
+    return f"Started timer for Q{qid+1}"
 
-# Submit response and record paradata
+# Submit response
 def submit_response(answer, qid):
-    end_time = time.time()
-    duration = end_time - paradata[qid]["start_time"]
-    paradata[qid]["response"] = answer
-    paradata[qid]["duration"] = duration
-    return f"Recorded Q{qid+1}: Time = {duration:.2f}s | Answer = {answer}"
+    if qid in paradata and "start_time" in paradata[qid]:
+        duration = time.time() - paradata[qid]["start_time"]
+        paradata[qid]["response"] = answer
+        paradata[qid]["duration"] = duration
+        return f"Q{qid+1} submitted in {duration:.2f} seconds. Answer: {answer}"
+    else:
+        return f"Timer not started for Q{qid+1}. Please type your answer to begin."
 
-# Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 AI Smart Survey Generator with Paradata")
+    gr.Markdown("## 🧠 AI Smart Survey Tool with Paradata Tracking")
 
-    prompt = gr.Textbox(label="🔤 Enter a survey topic prompt (e.g., Student Stress Levels)")
-    generate_btn = gr.Button("🚀 Generate Questions")
+    prompt = gr.Textbox(label="Enter survey topic prompt")
+    generate = gr.Button("Generate Survey Questions")
 
-    q_areas = []
-    responses = []
-    submits = []
+    qboxes = []
+    aboxes = []
+    submit_btns = []
+    logs = []
 
-    output_log = gr.Textbox(label="📝 System Log", interactive=False)
+    for i in range(5):
+        qbox = gr.Textbox(label=f"Q{i+1}", interactive=False)
+        abox = gr.Textbox(label=f"Answer Q{i+1}")
+        submit = gr.Button(f"Submit Q{i+1}")
+        log = gr.Textbox(label=f"Log Q{i+1}", interactive=False)
 
-    def display_questions(p):
-        questions = generate_questions(p)
-        for i, q in enumerate(questions):
-            q_box = gr.Textbox(value=q, label=f"Q{i+1}", interactive=False)
-            a_box = gr.Textbox(label=f"Answer Q{i+1}")
-            submit_btn = gr.Button(f"Submit Q{i+1}")
+        abox.change(start_timer, inputs=[gr.State(i)], outputs=log)
+        submit.click(submit_response, inputs=[abox, gr.State(i)], outputs=log)
 
-            a_box.change(start_timer, inputs=[gr.State(i)], outputs=output_log)
-            submit_btn.click(submit_response, inputs=[a_box, gr.State(i)], outputs=output_log)
+        qboxes.append(qbox)
+        aboxes.append(abox)
+        submit_btns.append(submit)
+        logs.append(log)
 
-            q_areas.append(q_box)
-            responses.append(a_box)
-            submits.append(submit_btn)
+    # Fill in all questions when prompt is submitted
+    def fill_questions(prompt_text):
+        questions = generate_questions(prompt_text)
+        return questions
 
-    generate_btn.click(display_questions, inputs=prompt, outputs=[])
+    generate.click(fill_questions, inputs=prompt, outputs=qboxes)
 
 demo.launch()