first working draft

Author: sunitj

README.md

@@ -1,2 +1,16 @@
 # LiteratureReviewBot
 Create a bot that scans pubmed literature and github for publications related to a topic and summarizes them in a table.
+
+## Dependencies
+
+You'll need to have an Ollama installed on your machine. You can download it by going [here](https://ollama.com/download). Once you have it installed you can download the LLama3.1 model by running the following command:
+
+```bash
+ollama pull llama3.1
+```
+
+Create a new virtual environment and install the dependencies by running the following commands:
+
+```bash
+pip install -eU .
+```

app.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_ollama.llms import OllamaLLM
+import streamlit as st
+
+from literaturereviewbot.prompts.literature_review import (
+    generate_prompt as lit_review_prompt,
+)
+
+st.title("Literature Review Bot")
+
+# Styling
+st.markdown(
+    """
+<style>
+.main {
+    background-color: #00000;
+}
+</style>
+""",
+    unsafe_allow_html=True,
+)
+
+# Sidebar for additional options or information
+with st.sidebar:
+    st.info("This app uses the Llama 3.1 model to answer your questions.")
+
+### Possible prompt for literature review bot once we have the data
+# documents = docs_to_be_summarized
+# # Construct the conversation
+
+
+###
+
+template = """{prompt}
+Answer:
+"""
+prompt = ChatPromptTemplate.from_template(template)
+model = OllamaLLM(model="llama3.1")
+chain = prompt | model
+
+# Main content
+col1, col2 = st.columns(2)
+with col1:
+    question = st.text_input("Enter your question here")
+    if question:
+        with st.spinner("Thinking..."):
+            prompt = lit_review_prompt(question)
+            answer = chain.invoke(prompt)
+            st.success("Done!")
+        st.markdown(f"**Answer:** {answer}")
+    else:
+        st.warning("Please enter a question to get an answer.")