Update app.py

app.py

@@ -1,4 +1,3 @@
-
 from openai import OpenAI
 import streamlit as st
 import os
@@ -8,6 +7,7 @@ import json
 import xml.etree.ElementTree as ET
 from io import StringIO, BytesIO
 from time import sleep
+from tabulate import tabulate
 
 # Load API key securely
 API_KEY = os.getenv("NV_API_KEY", "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")
@@ -43,47 +43,41 @@ def display_typing_animation(text):
         st.markdown(char, unsafe_allow_html=True)
         sleep(0.02)
 
-# Handle file uploads
+# Function to process uploaded files
 def process_uploaded_file(file):
     file_type = file.type
     if file_type == "text/csv":
         df = pd.read_csv(file)
-        return df.head().to_markdown()
+        return f"**CSV File Analysis:**\n{tabulate(df.head(), headers='keys', tablefmt='grid')}"
     elif file_type in ["application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", "application/vnd.ms-excel"]:
         df = pd.read_excel(file)
-        return df.head().to_markdown()
+        return f"**Excel File Analysis:**\n{tabulate(df.head(), headers='keys', tablefmt='grid')}"
     elif file_type == "application/json":
         data = json.load(file)
-        return json.dumps(data, indent=4)
-    elif file_type == "text/xml" or "application/xml":
+        return f"**JSON File Analysis:**\n```json\n{json.dumps(data, indent=4)}\n```"
+    elif file_type in ["text/xml", "application/xml"]:
         tree = ET.parse(file)
         root = tree.getroot()
-        return f"Root Tag: {root.tag}\nAttributes: {root.attrib}"
+        return f"**XML File Root Tag:** {root.tag}\n**Attributes:** {root.attrib}"
     elif file_type == "application/pdf":
-        return "PDF uploads are currently not processed for content. Try text-based analysis."
+        return "PDF uploads are not processed for content. Try text-based analysis."
     else:
-        return "Unsupported file format."
-
-# File upload section
-with st.sidebar.expander("Upload Files"):
-    uploaded_file = st.file_uploader("Upload your file (CSV, XLSX, JSON, XML, PDF)", type=["csv", "xlsx", "json", "xml", "pdf"])
-    if uploaded_file:
-        st.markdown(f"**Uploaded File Analysis:**\n```markdown\n{process_uploaded_file(uploaded_file)}\n```")
+        return "Unsupported file format. Please upload CSV, Excel, JSON, XML, or PDF files."
 
 # Display previous messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-# Handle user input
-if prompt := st.chat_input("Ask a coding-related question or general query"):
-    st.session_state.messages.append({"role": "user", "content": prompt})
+# Handle user input or file uploads
+if user_input := st.chat_input("Type a message or upload a file below."):
+    st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user"):
-        st.markdown(prompt)
+        st.markdown(user_input)
 
     # Assistant response
     with st.chat_message("assistant"):
-        with st.spinner('<div class="spinner">🌀 Thinking...</div>'):
+        with st.spinner("The assistant is processing..."):
             try:
                 # Generate response
                 stream = client.chat.completions.create(
@@ -100,7 +94,7 @@ if prompt := st.chat_input("Ask a coding-related question or general query"):
                         response_chunks.append(chunk.choices[0].delta.content)
                         display_typing_animation(chunk.choices[0].delta.content)
                 response = "".join(response_chunks)
-                st.markdown(f"```markdown\n{response}\n```")
+                st.markdown(response)
 
                 # Save assistant message
                 st.session_state.messages.append({"role": "assistant", "content": response})
@@ -108,16 +102,13 @@ if prompt := st.chat_input("Ask a coding-related question or general query"):
             except Exception as e:
                 st.error(f"An error occurred: {e}")
 
-# Add custom CSS for better visuals
-st.markdown("""
-<style>
-.spinner {
-    animation: pulse 1s infinite;
-}
-@keyframes pulse {
-    0% { opacity: 1; }
-    50% { opacity: 0.5; }
-    100% { opacity: 1; }
-}
-</style>
-""", unsafe_allow_html=True)
+# Add file upload to chat input
+st.markdown("### File Upload:")
+uploaded_file = st.file_uploader("Upload your file (CSV, XLSX, JSON, XML, PDF)", type=["csv", "xlsx", "json", "xml", "pdf"])
+if uploaded_file:
+    with st.chat_message("user"):
+        st.markdown(f"**Uploaded File:** {uploaded_file.name}")
+    with st.chat_message("assistant"):
+        with st.spinner("Processing the uploaded file..."):
+            file_analysis = process_uploaded_file(uploaded_file)
+            st.markdown(file_analysis)