Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,27 +1,87 @@
 import streamlit as st
-import os
-import json
-import sys
-from datetime import datetime, timedelta
-import base64
-import pandas as pd
-import pydeck as pdk
-import requests
 from paper import (
     literature_research_task, outline_task, draft_writing_task,
     citation_task, editing_task, chatbot_task,
     run_task
 )
 
+# Page config
+st.set_page_config(
+    page_title="📘 Academic Research Assistant",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# App title
+st.title("📘 Academic Research & Writing Assistant")
+st.markdown(
+    """
+    Welcome to your AI-powered academic assistant! Choose a task from the sidebar,
+    enter your topic and keywords, and let the agents do the rest.
+    """
+)
+
+# Sidebar task selection
+st.sidebar.header("🔧 Task Selector")
+task_choice = st.sidebar.radio(
+    "Select an academic task to perform:",
+    [
+        "Literature Review",
+        "Generate Paper Outline",
+        "Write Draft",
+        "Generate Citations",
+        "Edit Draft",
+        "Ask Chatbot"
+    ]
+)
+
+# Task mapping
+task_map = {
+    "Literature Review": literature_research_task,
+    "Generate Paper Outline": outline_task,
+    "Write Draft": draft_writing_task,
+    "Generate Citations": citation_task,
+    "Edit Draft": editing_task,
+    "Ask Chatbot": chatbot_task,
+}
 
-import ast  #추가 삽입, requirements: albumentations 추가
-script_repr = os.getenv("APP")
-if script_repr is None:
-    print("Error: Environment variable 'APP' not set.")
-    sys.exit(1)
+# Inputs
+st.subheader("🔍 Input Details")
 
-try:
-    exec(script_repr)
-except Exception as e:
-    print(f"Error executing script: {e}")
-    sys.exit(1)
+topic = st.text_input("Enter your research topic:", placeholder="e.g. Impact of AI in Healthcare")
+keywords = st.text_input("Enter related keywords (comma-separated):", placeholder="e.g. AI, healthcare, diagnosis")
+
+# Optional input for chatbot queries
+query = ""
+if task_choice == "Ask Chatbot":
+    query = st.text_area("Ask your question:", placeholder="e.g. How to format a research abstract?")
+
+# Action button
+if st.button("🚀 Run Agent"):
+    if not topic and task_choice != "Ask Chatbot":
+        st.warning("Please enter a topic for the selected task.")
+    elif task_choice == "Ask Chatbot" and not query:
+        st.warning("Please enter a chatbot query.")
+    else:
+        with st.spinner("Agent is working on your task..."):
+            # Format input text based on task type
+            input_text = (
+                query if task_choice == "Ask Chatbot"
+                else f"Topic: {topic}\nKeywords: {keywords}"
+            )
+            result = run_task(task_map[task_choice], input_text)
+
+        st.success("✅ Task Completed!")
+        st.subheader("📄 Result")
+        st.markdown(result, unsafe_allow_html=True)
+
+# Footer
+st.markdown(
+    """
+    <hr>
+    <center>
+        <small>Powered by LangChain + Gemini | Built by Academic AI</small>
+    </center>
+    """,
+    unsafe_allow_html=True
+)