Update app.py

app.py

@@ -1,13 +1,4 @@
-from langchain_community.chat_message_histories import StreamlitChatMessageHistory
 import streamlit as st
-from langchain.prompts import (
-    ChatPromptTemplate,
-    HumanMessagePromptTemplate,
-    MessagesPlaceholder,
-)
-from more_itertools import chunked
-
-from langserve import RemoteRunnable
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import os
 from langchain import PromptTemplate
@@ -15,17 +6,17 @@ from langchain import LLMChain
 from langchain_together import Together
 import re
 import pdfplumber
-# Set the API key with double quotes
 
+# Set the API key
 os.environ['TOGETHER_API_KEY'] = "5653bbfbaf1f7c1438206f18e5dfc2f5992b8f0b6aa9796b0131ea454648ccde"
 
 text = ""
 max_pages = 16
 with pdfplumber.open("AI Engineer Test.pdf") as pdf:
-        for i, page in enumerate(pdf.pages):
-            if i >= max_pages:
-                break
-            text += page.extract_text() + "\n"
+    for i, page in enumerate(pdf.pages):
+        if i >= max_pages:
+            break
+        text += page.extract_text() + "\n"
 
 def Bot(Questions):
     chat_template = """
@@ -40,52 +31,42 @@ def Bot(Questions):
     llama3 = Together(model="meta-llama/Llama-3-70b-chat-hf", max_tokens=250)
     Generated_chat = LLMChain(llm=llama3, prompt=prompt)
 
-   
+    try:
+        response = Generated_chat.invoke({
+            "text": text,
+            "Questions": Questions
+        })
+
+        response_text = response['text']
+
+        response_text = response_text.replace("assistant", "")
+
+        # Post-processing to handle repeated words and ensure completeness
+        words = response_text.split()
+        seen = set()
+        filtered_words = [word for word in words if word.lower() not in seen and not seen.add(word.lower())]
+        response_text = ' '.join(filtered_words)
+        response_text = response_text.strip()  # Ensuring no extra spaces at the ends
+        if not response_text.endswith('.'):
+            response_text += '.'
+
+        return response_text
+    except Exception as e:
+        return f"Error in generating response: {e}"
+
 def ChatBot(Questions):
-  greetings = ["hi", "hello", "hey", "greetings", "what's up", "howdy"]
+    greetings = ["hi", "hello", "hey", "greetings", "what's up", "howdy"]
     # Check if the input question is a greeting
-  question_lower = Questions.lower().strip()
-  if question_lower in greetings or any(question_lower.startswith(greeting) for greeting in greetings):
+    question_lower = Questions.lower().strip()
+    if question_lower in greetings or any(question_lower.startswith(greeting) for greeting in greetings):
         return "Hello! How can I assist you with the document today?"
-  else:
-    response=Bot(Questions)
-    return response.translate(str.maketrans('', '', '\n'))
-"""
-  # --- Logo ---
-st.set_page_config(
-    page_title="AI Engineer Test Chatbot",
-    page_icon="Insight Therapy Solutions.png",
-    layout="wide",
-)
-st.sidebar.image("Insight Therapy Solutions.png", width=200)
+    else:
+        response = Bot(Questions)
+        return response.translate(str.maketrans('', '', '\n'))
 
-st.sidebar.title("Navigation")
-st.sidebar.write("Reclaim Your Mental Health")
-st.sidebar.markdown("[Visit us at](https://www.insighttherapysolutions.com/)")
-"""
-# Add some custom styling
-st.markdown(
-    
-    <style>
-    .css-18e3th9 {
-        padding-top: 3rem;
-    }
-    .css-1d391kg {
-        text-align: center;
-    }
-    .stButton>button {
-        background-color: #4CAF50;
-        color: white;
-        border: none;
-        padding: 15px 32px;
-        text-align: center;
-        text-decoration: none;
-        display: inline-block;
-        font-size: 16px;
-        margin: 4px 2px;
-        cursor: pointer;
-        border-radius: 8px;
-    }
-    </style>
-    , unsafe_allow_html=True
-)
+# Streamlit UI
+st.title("Chatbot")
+Questions = st.text_input("Ask a question:")
+if st.button("Submit"):
+    answer = ChatBot(Questions)
+    st.write(answer)