Update app.py

app.py

@@ -24,22 +24,11 @@ def extract_pdf_text(uploaded_file):
         pdf_text += page.extract_text()
     return pdf_text
 
-def add_message_to_conversation(user_message, bot_message, model_name):
-    if "conversation" not in st.session_state:
-        st.session_state.conversation = []
-    st.session_state.conversation.append((user_message, bot_message, model_name))
-
 # Streamlit app
 st.set_page_config(page_title="Gemma 27B-it Chatbot Interface", layout="wide")
 st.title("Gemma 27B-it Chatbot Interface")
 st.write("Gemma 27B-it Chatbot Interface")
 
-# Initialize session state for conversation and uploaded file
-if "conversation" not in st.session_state:
-    st.session_state.conversation = []
-if "uploaded_file" not in st.session_state:
-    st.session_state.uploaded_file = None
-
 # File uploader for PDF
 uploaded_file = st.file_uploader("Upload a PDF", type="pdf")
 
@@ -50,28 +39,15 @@ if uploaded_file:
     st.write(pdf_text)
 
 # User input for question
-question = st.text_input("Question", placeholder="Enter your question here...")
+question = st.text_input("Ask a question...", "")
 
 # Handle user input and Gemma 27B-it model response
 if st.button("Send") and question:
     try:
         with st.spinner("Waiting for the model to respond..."):
-            # Construct the chat history
-            chat_history = " ".join([msg[1] for msg in st.session_state.conversation[-5:]]) + f"User: {question}\n"
-            response = query_model(GEMMA_27B_API_URL, {"inputs": chat_history})
-
-            if isinstance(response, list):
-                answer = response[0].get("generated_text", "No response")
-            elif isinstance(response, dict):
-                answer = response.get("generated_text", "No response")
-            else:
-                answer = "No response"
-
-            # Add PDF text to the chat history
-            if st.session_state.uploaded_file:
-                chat_history += f"Document Text: {pdf_text}\n"
-
-            add_message_to_conversation(question, answer, "Gemma-2-27B-it")
+            response = query_model(GEMMA_27B_API_URL, {"inputs": question})
+            answer = response.get("generated_text", "No response")
+            st.write(f"**Gemma 27B-it:** {answer}")
     except ValueError as e:
         st.error(str(e))
 
@@ -79,36 +55,41 @@ if st.button("Send") and question:
 st.markdown(
     """
     <style>
-    .chat-bubble {
-        padding: 10px 14px;
-        border-radius: 14px;
-        margin-bottom: 10px;
-        display: inline-block;
-        max-width: 80%;
-        color: black;
-    }
-    .chat-bubble.user {
-        background-color: #dcf8c6;
-        align-self: flex-end;
-    }
-    .chat-bubble.bot {
-        background-color: #fff;
-        align-self: flex-start;
-    }
     .chat-container {
         display: flex;
         flex-direction: column;
         gap: 10px;
         margin-top: 20px;
     }
+    .user-message {
+        align-self: flex-end;
+        background-color: #dcf8c6;
+        padding: 10px 14px;
+        border-radius: 14px;
+        max-width: 80%;
+    }
+    .bot-message {
+        align-self: flex-start;
+        background-color: #fff;
+        padding: 10px 14px;
+        border-radius: 14px;
+        max-width: 80%;
+    }
     </style>
     """,
     unsafe_allow_html=True
 )
 
 # Display the conversation
-st.write('<div class="chat-container">', unsafe_allow_html=True)
-for user_message, bot_message, model_name in st.session_state.conversation:
-    st.write(f'<div class="chat-bubble user">You: {user_message}</div>', unsafe_allow_html=True)
-    st.write(f'<div class="chat-bubble bot">{model_name}: {bot_message}</div>', unsafe_allow_html=True)
-st.write('</div>', unsafe_allow_html=True)
+if st.session_state.conversation:
+    st.write('<div class="chat-container">', unsafe_allow_html=True)
+    for user_message, bot_message in st.session_state.conversation:
+        st.write(f'<div class="user-message">You: {user_message}</div>', unsafe_allow_html=True)
+        st.write(f'<div class="bot-message">Gemma 27B-it: {bot_message}</div>', unsafe_allow_html=True)
+    st.write('</div>', unsafe_allow_html=True)
+
+# Add the current interaction to the conversation
+if question:
+    st.session_state.conversation.append((question, answer))
+else:
+    st.session_state.conversation = []