| import streamlit as st |
| import os |
| import sys |
| from typing import Tuple, List, Dict, Optional |
| from router import Router |
|
|
| sys.path.append(os.path.dirname(os.path.abspath(__file__))) |
|
|
| class AcademicResearchAssistant: |
| def __init__(self): |
| |
| self.router = Router() |
| self.setup_streamlit_config() |
| self.initialize_session_state() |
|
|
| def setup_streamlit_config(self): |
| |
| st.set_page_config( |
| page_title="Academic Research Assistant", |
| page_icon="📚", |
| layout="wide", |
| initial_sidebar_state="expanded", |
| menu_items={ |
| 'Get Help': 'https://github.com/yourusername/academic-research-assistant', |
| 'Report a bug': "https://github.com/yourusername/academic-research-assistant/issues", |
| 'About': "# Academic Research Assistant v1.0\nYour intelligent research companion." |
| } |
| ) |
|
|
| |
| st.markdown(""" |
| <style> |
| .stApp { |
| background: linear-gradient(to bottom right, #f5f7fa, #eef2f7); |
| } |
| .stButton>button { |
| background-color: #1f4287; |
| color: white; |
| border-radius: 5px; |
| padding: 0.5rem 1rem; |
| } |
| .stProgress .st-bo { |
| background-color: #1f4287; |
| } |
| .chat-message { |
| padding: 10px; |
| border-radius: 5px; |
| margin: 5px 0; |
| animation: fadeIn 0.5s ease-in; |
| } |
| .user-message { |
| background-color: #e6f3ff; |
| } |
| .bot-message { |
| background-color: #f0f2f6; |
| } |
| @keyframes fadeIn { |
| from {opacity: 0;} |
| to {opacity: 1;} |
| } |
| .paper-card { |
| background-color: white; |
| padding: 1.5rem; |
| border-radius: 10px; |
| box-shadow: 0 2px 4px rgba(0,0,0,0.1); |
| margin-bottom: 1rem; |
| } |
| .paper-title { |
| color: #1f4287; |
| font-size: 1.1rem; |
| font-weight: bold; |
| margin-bottom: 0.5rem; |
| } |
| .paper-metadata { |
| font-size: 0.9rem; |
| color: #666; |
| margin-bottom: 0.5rem; |
| } |
| .paper-abstract { |
| font-size: 0.95rem; |
| line-height: 1.5; |
| margin-top: 1rem; |
| padding-left: 1rem; |
| border-left: 3px solid #1f4287; |
| } |
| .download-button { |
| background-color: #4CAF50; |
| color: white; |
| padding: 0.5rem 1rem; |
| border-radius: 5px; |
| text-decoration: none; |
| display: inline-block; |
| margin-top: 1rem; |
| } |
| .metric-card { |
| background-color: white; |
| padding: 1rem; |
| border-radius: 8px; |
| box-shadow: 0 2px 4px rgba(0,0,0,0.05); |
| text-align: center; |
| } |
| </style> |
| """, unsafe_allow_html=True) |
|
|
| def initialize_session_state(self): |
| |
| if "chat_history" not in st.session_state: |
| st.session_state.chat_history = [] |
| if "fetched_papers" not in st.session_state: |
| st.session_state.fetched_papers = [] |
| if "search_count" not in st.session_state: |
| st.session_state.search_count = 0 |
| if "total_searches" not in st.session_state: |
| st.session_state.total_searches = 0 |
|
|
| def display_welcome_message(self): |
|
|
| st.title("📚 Academic Research Paper Assistant") |
| |
| |
| col1, col2, col3, col4 = st.columns([2, 1, 1, 1]) |
| |
| with col1: |
| st.markdown(""" |
| Welcome to your intelligent research companion! This tool helps you: |
| - 🔍 Find relevant academic papers |
| - 📊 Analyze research trends |
| - 📖 Access paper summaries |
| - 📥 Download full papers |
| """) |
| |
| |
| with col3: |
| st.markdown(""" |
| <div class="metric-card"> |
| <h3>Papers Found</h3> |
| <h2>{}</h2> |
| </div> |
| """.format(len(st.session_state.fetched_papers)), unsafe_allow_html=True) |
| |
| with col4: |
| st.markdown(""" |
| <div class="metric-card"> |
| <h3>Total Searches</h3> |
| <h2>{}</h2> |
| </div> |
| """.format(st.session_state.total_searches), unsafe_allow_html=True) |
| |
| |
|
|
| def create_chat_interface(self) -> Tuple[str, bool]: |
| |
| with st.container(): |
| st.write("### 💬 Research Query Interface") |
| |
| |
| col1, col2 = st.columns([4, 1]) |
| |
| with col1: |
| user_input = st.text_input( |
| "Enter your research query (e.g., 'Recent advances in quantum computing')", |
| key="user_input", |
| placeholder="Type your research question here...", |
| max_chars=500 |
| ) |
| |
| col3, col4, col5 = st.columns([2, 1, 1]) |
| with col3: |
| send_button = st.button("🔍 Search ", use_container_width=True) |
| with col4: |
| clear_button = st.button("🗑️ Clear History", use_container_width=True) |
| |
| if clear_button: |
| st.session_state.chat_history = [] |
| st.session_state.fetched_papers = [] |
| st.session_state.search_count = 0 |
| st.session_state.total_searches = 0 |
| st.rerun() |
| |
| return user_input, send_button |
|
|
| def process_user_input(self, user_input: str): |
| |
| with st.spinner('🔍 Working on response...'): |
| |
| st.session_state.search_count = len(st.session_state.fetched_papers) |
| st.session_state.total_searches += 1 |
| |
| try: |
| |
| response, papers = self.router.route_query(user_input) |
| |
| |
| if papers: |
| unique_papers = {paper['paper_number']: paper for paper in papers} |
| st.session_state.fetched_papers = list(unique_papers.values()) |
| |
| |
| if response: |
| st.session_state.chat_history.append(("Bot", response)) |
| st.session_state.chat_history.append(("User", user_input)) |
| else: |
| st.session_state.chat_history.append( |
| ("Bot", ["I couldn't find relevant papers for your query. Please try rephrasing or use more specific terms."]) |
| ) |
| except Exception as e: |
| st.session_state.chat_history.append( |
| ("Bot", [f"An error occurred while processing your request: {str(e)}"]) |
| ) |
| st.error("There was an error processing your request. Please try again.") |
|
|
| def display_chat_history(self): |
| """Display the chat history with user and bot messages""" |
| for sender, message in reversed(st.session_state.chat_history): |
| if sender == "User": |
| st.markdown( |
| "<div class='chat-message user-message'>" |
| f"<strong>👤 You:</strong> {message}" |
| "</div>", |
| unsafe_allow_html=True |
| ) |
| else: |
| st.markdown( |
| "<div class='chat-message bot-message'>" |
| f"<strong>🤖 Assistant:</strong> {message[0]}" |
| "</div>", |
| unsafe_allow_html=True |
| ) |
|
|
| def display_papers(self): |
| """Display the list of fetched papers with download links""" |
| st.write("### 📄 Retrieved Research Papers") |
| if st.session_state.fetched_papers: |
| for paper in st.session_state.fetched_papers: |
| with st.expander(f"📑 {paper.get('title', 'Untitled Paper')}"): |
| st.markdown( |
| """ |
| <div class='paper-card'> |
| <div class='paper-title'>{}</div> |
| <div class='paper-metadata'> |
| Year: {} | Paper ID: {} |
| </div> |
| {} |
| </div> |
| """.format( |
| paper.get('title', '').replace('\n', ' ').strip(), |
| paper.get('year', 'N/A'), |
| paper.get('paper_number', 'N/A'), |
| '<div class="paper-abstract">' + paper.get('abstract', '') + '</div>' if paper.get('abstract') else '' |
| ), |
| unsafe_allow_html=True |
| ) |
|
|
|
|
| |
| download_link = paper.get('link') |
| if download_link: |
| st.markdown(f"[📥 Download PDF]({download_link})") |
| else: |
| st.warning("⚠️ No download link available") |
| else: |
| st.info("🔍 No papers fetched yet. Start by entering a research query above!") |
|
|
| def run(self): |
| """Main method to run the application""" |
| self.display_welcome_message() |
| |
|
|
| user_input, send_button = self.create_chat_interface() |
|
|
| st.markdown("### 💬 Chat History") |
| self.display_chat_history() |
| |
|
|
| if user_input and send_button: |
| self.process_user_input(user_input) |
| st.rerun() |
| |
| st.markdown("---") |
| self.display_papers() |
|
|
| def main(): |
|
|
| app = AcademicResearchAssistant() |
| app.run() |
|
|
| if __name__ == "__main__": |
| main() |
