Update app.py

app.py

@@ -15,7 +15,10 @@ HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
 def query_model(api_url, payload):
     response = requests.post(api_url, headers=HEADERS, json=payload)
-    return response.json()
+    if response.status_code == 200:
+        return response.json()
+    else:
+        raise ValueError(f"Request failed with status code {response.status_code}: {response.text}")
 
 def extract_pdf_text(uploaded_file):
     pdf_text = ""
@@ -29,6 +32,10 @@ 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 = []
+
 # File uploader for PDF
 uploaded_file = st.file_uploader("Upload a PDF", type="pdf")
 
@@ -48,6 +55,7 @@ if st.button("Send") and question:
             response = query_model(GEMMA_27B_API_URL, {"inputs": question})
             answer = response.get("generated_text", "No response")
             st.write(f"**Gemma 27B-it:** {answer}")
+            st.session_state.conversation.append((question, answer))
     except ValueError as e:
         st.error(str(e))
 
@@ -86,10 +94,4 @@ if st.session_state.conversation:
     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 = []
+    st.write('</div>', unsafe_allow_html=True)