Update pages/summarizer.py

pages/summarizer.py

@@ -17,19 +17,23 @@ def query_huggingface(payload):
         st.error(f"Error querying Hugging Face API: {e}")
         return {"summary_text": f"Error querying Hugging Face API: {e}"}
 
-
 def main():
-    
-    st.set_page_config(layout="centered")
+    st.set_page_config(layout="wide")
     st.title("Chat Summarizer")
 
+    # Initialize a list to store chat messages
+    chat_history = []
+
     # User input for chat message
-    user_message = st.text_input("User Message", "Enter your message here...")
+    user_message = st.text_input("Provide a Chat/Long description to summarize")
 
-    # Process user input and query Hugging Face API
-    if st.button("Summarize"):
+    # Process user input and query Hugging Face API on button click
+    if st.button("Send"):
         if user_message:
-            # Construct input text for summarization (no system message)
+            # Add user message to chat history
+            chat_history.append({"speaker": "User", "message": user_message})
+
+            # Construct input text for summarization
             input_text = f"User: {user_message}"
 
             # Query Hugging Face API for summarization
@@ -37,14 +41,17 @@ def main():
             response = query_huggingface(payload)
 
             # Extract summary text from the API response
-            summary_text = response[0]["summary_text"] if isinstance(response, list) else response.get("summary_text", "")
+            summary_text = response.get("summary_text", "")
 
-            # Display summary text
-            st.text_area("Summary", value=summary_text)
+            # Add summarization response to chat history
+            chat_history.append({"speaker": "Bot", "message": summary_text})
 
-    with st.sidebar:        
-        if st.button("Chat With PDF"):
-            st.switch_page('app.py')
+    # Display chat history as a conversation
+    for chat in chat_history:
+        if chat["speaker"] == "User":
+            st.text_input("User", chat["message"], disabled=True)
+        elif chat["speaker"] == "Bot":
+            st.text_area("Bot", chat["message"], disabled=True)
 
 if __name__ == "__main__":
     main()