Create app.py

app.py

@@ -0,0 +1,90 @@
+from transformers import pipeline
+import gradio as gr
+
+# Step 1: Load model pipeline
+generator = pipeline(
+    "text-generation",
+    model="HuggingFaceH4/zephyr-7b-alpha",
+    max_new_tokens=512,
+    temperature=0.7,
+    do_sample=True,
+    device_map="auto"
+)
+
+# Step 2: Agent functions
+def chat_with_agent(role_description, prompt):
+    system = f"[System: {role_description}]\n"
+    input_text = system + prompt
+    response = generator(input_text, return_full_text=False)[0]['generated_text']
+    return response.replace(input_text, "").strip()
+
+def get_topic():
+    return chat_with_agent("You are a task manager deciding AI research paper topics.",
+                           "Suggest a trending topic in AI for academic writing.")
+
+def get_research(topic):
+    return chat_with_agent("You are a research assistant gathering key points.",
+                           f"Give me 5 major points about: {topic}")
+
+def generate_draft(research_points):
+    return chat_with_agent("You are a technical writer.",
+                           f"Write a short 300-word academic draft based on:\n{research_points}")
+
+def review_draft(draft):
+    return chat_with_agent("You are a senior editor. Check for grammar, clarity, and flow.",
+                           f"Please review and improve the following draft:\n\n{draft}")
+
+def format_final(reviewed):
+    return chat_with_agent("You are a publisher formatting for IEEE conference style.",
+                           f"Format this content:\n{reviewed}")
+
+# Step 3: Gradio function for full pipeline
+def multi_agent_ui(topic_choice=None):
+    topic = topic_choice if topic_choice else get_topic()
+    research = get_research(topic)
+    draft = generate_draft(research)
+    reviewed = review_draft(draft)
+    final_output = format_final(reviewed)
+
+    return {
+        "Topic": topic,
+        "Research Points": research,
+        "Draft": draft,
+        "Reviewed Draft": reviewed,
+        "Final Formatted Output": final_output
+    }
+
+# Step 4: Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 AI Research Paper Assistant (Multi-Agent System)")
+    
+    with gr.Row():
+        topic_input = gr.Textbox(label="Enter Topic (Optional)", placeholder="Leave blank to auto-generate")
+        generate_btn = gr.Button("Generate Research Paper")
+    
+    with gr.Accordion("📌 Topic", open=False):
+        topic_output = gr.Textbox(label="Topic", lines=2)
+    
+    with gr.Accordion("📚 Research Points", open=False):
+        research_output = gr.Textbox(label="Research Points", lines=6)
+    
+    with gr.Accordion("📝 Draft", open=False):
+        draft_output = gr.Textbox(label="Draft", lines=10)
+    
+    with gr.Accordion("✅ Reviewed", open=False):
+        reviewed_output = gr.Textbox(label="Reviewed Draft", lines=10)
+    
+    with gr.Accordion("📄 Final Paper", open=True):
+        final_output = gr.Textbox(label="Formatted Final Output", lines=12)
+
+    generate_btn.click(
+        fn=multi_agent_ui,
+        inputs=[topic_input],
+        outputs=[
+            topic_output, research_output,
+            draft_output, reviewed_output,
+            final_output
+        ]
+    )
+
+demo.launch()