Update app.py

app.py

@@ -16,15 +16,23 @@ TEMPERATURE = 0.1
 TOP_P = 0.9
 
 @spaces.GPU
-def respond_stream(title, abstract):
-    if not title.strip() or not abstract.strip():
-        return "❌ Error: Title and Abstract are required."
+def respond_stream(summary, title, abstract):
+    # Validate mandatory fields
+    if not summary.strip() or not title.strip() or not abstract.strip():
+        return "❌ Error: PICOS Summary, Title, and Abstract are all required."
 
-    prompt = f"Title: {title.strip()}\nAbstract: {abstract.strip()}"
+    # Build prompt
+    prompt = (
+        f"Instruction: Use the following PICOS summary to evaluate the abstract.\n"
+        f"\nPICOS Summary: {summary.strip()}"
+        f"\n\nTitle: {title.strip()}\nAbstract: {abstract.strip()}"
+    )
 
+    # Wrap into message for chat template
     messages = [{"role": "user", "content": prompt}]
     prompt_text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
+    # Tokenize and prepare streamer
     inputs = tokenizer(prompt_text, return_tensors="pt").to("cuda")
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
@@ -48,9 +56,10 @@ def respond_stream(title, abstract):
 
 # Build Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Qwen Streaming Chat — Medical Abstract Evaluator")
+    gr.Markdown("## 🤖 Qwen Streaming Chat — Abstract Evaluation with PICOS Summary")
 
     with gr.Column():
+        summary = gr.Textbox(label="PICOS Summary", lines=4, placeholder="Required")
         title = gr.Textbox(label="Title", lines=2, placeholder="Required")
         abstract = gr.Textbox(label="Abstract", lines=10, placeholder="Required")
 
@@ -59,7 +68,7 @@ with gr.Blocks() as demo:
 
     generate_btn.click(
         fn=respond_stream,
-        inputs=[title, abstract],
+        inputs=[summary, title, abstract],
         outputs=[output_box]
     )