add manual input

app.py

@@ -12,7 +12,7 @@ from transformers import pipeline
 import asyncio
 
 # Import the required functions from the pipeline file
-from pipeline_gradio_experimental import generate_basic_question, rank_questions_with_details
+from pipeline_gradio_experimental import generate_single_question, rank_questions_with_details
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -85,68 +85,72 @@ Score each question-answer pair on a scale of 0 to 10 based on the quality and r
         logger.error(f"Error in comparing questions: {e}")
         return {"question1_score": 0, "question2_score": 0, "explanation": "Failed to compare questions"}
 
-async def process_random_entry(progress=gr.Progress()):
-    context, original_answer, original_question = get_random_entry()
-    
-    # Yield the original context, question, and answer immediately
-    yield context, original_question, original_answer, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
-    
-    # Simulate some processing time
+async def process_entry(context, answer, progress=gr.Progress()):
+    progress(0, desc="Starting process...")
     await asyncio.sleep(1)
-    progress(0.3, desc="Generating questions...")
     
-    basic_question = generate_basic_question(context, original_answer)
-    _, _, enhanced_question = rank_questions_with_details(context, original_answer)
-    
-    await asyncio.sleep(1)
-    progress(0.6, desc="Generating answers...")
+    progress(0.2, desc="Generating questions...")
+    basic_question = generate_single_question(context, answer, [])
+    _, _, enhanced_question = rank_questions_with_details(context, answer)
     
+    progress(0.4, desc="Generating answers...")
     basic_answer = generate_answer(context, basic_question)
     enhanced_answer = generate_answer(context, enhanced_question)
     
-    await asyncio.sleep(1)
-    progress(0.9, desc="Comparing questions...")
-    
-    comparison_result = compare_questions(context, original_answer, basic_question, basic_answer, enhanced_question, enhanced_answer)
+    progress(0.6, desc="Comparing questions...")
+    comparison_result = compare_questions(context, answer, basic_question, basic_answer, enhanced_question, enhanced_answer)
     
     winner = "Basic" if comparison_result["question1_score"] > comparison_result["question2_score"] else "Enhanced"
     
-    # Yield the final results
-    yield (
-        context,
-        original_question,
-        original_answer,
-        gr.update(visible=True),
-        gr.update(visible=True, value=f"Question: {basic_question}\nAnswer: {basic_answer}"),
-        gr.update(visible=True, value=f"Question: {enhanced_question}\nAnswer: {enhanced_answer}"),
-        gr.update(visible=True, value=f"Question 1 Score: {comparison_result['question1_score']}\n"
-                                      f"Question 2 Score: {comparison_result['question2_score']}\n"
-                                      f"Explanation: {comparison_result['explanation']}\n"
-                                      f"Winner: {winner} Generation")
+    progress(1.0, desc="Process complete!")
+    return (
+        f"Question: {basic_question}\nAnswer: {basic_answer}",
+        f"Question: {enhanced_question}\nAnswer: {enhanced_answer}",
+        f"Question 1 Score: {comparison_result['question1_score']}\n"
+        f"Question 2 Score: {comparison_result['question2_score']}\n"
+        f"Explanation: {comparison_result['explanation']}\n"
+        f"Winner: {winner} Generation"
     )
 
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Default()) as iface:
     gr.Markdown("# Question Generation and Comparison")
-    gr.Markdown("Click the button to get a random entry from the SQuAD dataset and compare basic and enhanced question generation.")
+    gr.Markdown("Enter a context and answer, or click 'Random' to get a random entry from the SQuAD dataset.")
     
-    random_button = gr.Button("Get Random Question")
-    
-    with gr.Column(visible=False) as output_column:
-        context_output = gr.Textbox(label="Original Context")
-        original_question_output = gr.Textbox(label="Original Question")
-        original_answer_output = gr.Textbox(label="Original Answer")
-        basic_generation_output = gr.Textbox(label="Basic Generation", visible=False)
-        enhanced_generation_output = gr.Textbox(label="Enhanced Generation", visible=False)
-        comparison_result_output = gr.Textbox(label="Comparison Result", visible=False)
+    with gr.Row():
+        with gr.Column(scale=2):
+            context_input = gr.Textbox(label="Context", lines=10)
+            answer_input = gr.Textbox(label="Answer", lines=2)
+            with gr.Row():
+                submit_button = gr.Button("Submit")
+                random_button = gr.Button("Random")
+        
+        with gr.Column(scale=3):
+            original_question_output = gr.Textbox(label="Original Question from Dataset", lines=2)
+            basic_generation_output = gr.Textbox(label="Basic Generation", lines=4)
+            enhanced_generation_output = gr.Textbox(label="Enhanced Generation", lines=4)
+            comparison_result_output = gr.Textbox(label="Comparison Result", lines=6)
+
+    async def on_submit(context, answer):
+        return await process_entry(context, answer)
+
+    async def on_random():
+        context, answer, question = get_random_entry()
+        results = await process_entry(context, answer)
+        return [context, answer, question] + list(results)
+
+    submit_button.click(
+        fn=on_submit,
+        inputs=[context_input, answer_input],
+        outputs=[basic_generation_output, enhanced_generation_output, comparison_result_output]
+    )
 
     random_button.click(
-        fn=process_random_entry,
+        fn=on_random,
         outputs=[
-            context_output,
+            context_input,
+            answer_input,
             original_question_output,
-            original_answer_output,
-            output_column,
             basic_generation_output,
             enhanced_generation_output,
             comparison_result_output