Update app.py

app.py

@@ -1,7 +1,10 @@
-BACKEND_URL = "https://huggingface.co/spaces/shiva9876/code_gen_backend"
 import streamlit as st
 import requests
 import re
+import uuid
+
+# Backend URL - Update this with your actual Hugging Face Space URL
+BACKEND_URL = "https://huggingface.co/spaces/shiva9876/code_gen_backend"
 
 # Page Config with Logo
 st.set_page_config(
@@ -90,6 +93,20 @@ st.markdown("""
         .search-button:hover {
             background-color: #02c2ad;
         }
+
+        /* Clear Chat Button */
+        .stButton>button {
+            background-color: #ff4444;
+            color: white;
+            border-radius: 5px;
+            padding: 5px 15px;
+            border: none;
+            cursor: pointer;
+        }
+
+        .stButton>button:hover {
+            background-color: #cc0000;
+        }
     </style>
 """, unsafe_allow_html=True)
 
@@ -102,26 +119,71 @@ st.markdown(
 # Title
 st.markdown('<div class="title">Write any code: Verbose❇️</div>', unsafe_allow_html=True)
 
-# Chat history
+# Initialize session state
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
+if "session_id" not in st.session_state:
+    st.session_state.session_id = str(uuid.uuid4())
+
+# Display chat history
 for msg in st.session_state.messages:
     if msg["role"] == "user":
-        st.markdown(f'<div class="chat-container user-message"><b>User:</b><br>{msg["content"]}</div>',
+        st.markdown(f'<div class="chat-container user-message"><b>You:</b><br>{msg["content"]}</div>',
                     unsafe_allow_html=True)
     elif msg["role"] == "ai":
-        explanation, code = msg["content"]
-        st.markdown(f'<div class="chat-container ai-message"><b>AI:</b><br>{explanation}</div>', unsafe_allow_html=True)
-        if code:
-        #     st.subheader("Generated Code:")
-            st.code(code, language="python")
+        content = msg["content"]
+        
+        # Check if it's a conversational response or code response
+        if isinstance(content, tuple):
+            explanation, code = content
+            if explanation:
+                st.markdown(f'<div class="chat-container ai-message"><b>AI:</b><br>{explanation}</div>', 
+                           unsafe_allow_html=True)
+            if code:
+                st.code(code, language="python")
+        else:
+            # Simple text response (for conversational messages)
+            st.markdown(f'<div class="chat-container ai-message"><b>AI:</b><br>{content}</div>', 
+                       unsafe_allow_html=True)
+
+# Sidebar with options
+with st.sidebar:
+    st.header("Options")
+    
+    response_type = st.selectbox(
+        "Response Type:",
+        ["both", "code", "explanation", "conversation"],
+        help="Choose what type of response you want"
+    )
+    
+    if st.button("🗑️ Clear Chat History"):
+        try:
+            # Clear on backend
+            requests.post(
+                f"{BACKEND_URL}/clear_history/",
+                json={"session_id": st.session_state.session_id}
+            )
+        except:
+            pass
+        
+        # Clear locally
+        st.session_state.messages = []
+        st.session_state.session_id = str(uuid.uuid4())
+        st.rerun()
+    
+    st.divider()
+    st.caption("💡 Tip: Ask coding questions or request code generation!")
 
 # Input Form
 st.markdown('<div class="search-container">', unsafe_allow_html=True)
 
 with st.form(key="input_form"):
-    user_input = st.text_input("Your Prompt:", key="user_prompt", placeholder="Enter your prompt...")
+    user_input = st.text_input(
+        "Your Prompt:", 
+        key="user_prompt", 
+        placeholder="e.g., Write a function to sort a list or just say Hi!"
+    )
 
     # Single send button
     submit_button = st.form_submit_button("SEND ↗️")
@@ -131,25 +193,96 @@ st.markdown("</div>", unsafe_allow_html=True)
 # API Call Logic
 if submit_button:
     if user_input.strip():
+        # Add user message to chat
         st.session_state.messages.append({"role": "user", "content": user_input})
-        try:
-            response = requests.post(f"{BACKEND_URL}/generate/", json={"prompt": user_input, "response_type": "both"})
-            if response.status_code == 200:
-                data = response.json()
-                raw_text = data.get("response", "").strip()
-                if not raw_text or raw_text.lower() == "none":
-                    st.warning("⚠️ No valid response generated. Try a different prompt.")
-                else:
-                    explanation, code = raw_text, None
-                    code_match = re.search(r"```(?:python)?\n(.*?)\n```", raw_text, re.DOTALL)
-                    if code_match:
-                        code = code_match.group(1).strip()
-                        explanation = raw_text.replace(code_match.group(0), "").strip()
-                    st.session_state.messages.append({"role": "ai", "content": (explanation, code)})
+        
+        # Show loading spinner
+        with st.spinner("🤔 Thinking..."):
+            try:
+                # Make API call to backend
+                response = requests.post(
+                    f"{BACKEND_URL}/generate/",
+                    json={
+                        "prompt": user_input,
+                        "session_id": st.session_state.session_id,
+                        "response_type": response_type
+                    },
+                    timeout=60
+                )
+                
+                if response.status_code == 200:
+                    data = response.json()
+                    
+                    # Update session_id if returned
+                    if "session_id" in data:
+                        st.session_state.session_id = data["session_id"]
+                    
+                    message_type = data.get("message_type", "code")
+                    
+                    if message_type == "conversation":
+                        # Handle conversational response
+                        ai_response = data.get("response", "")
+                        if ai_response and ai_response.strip().lower() != "none":
+                            st.session_state.messages.append({
+                                "role": "ai", 
+                                "content": ai_response
+                            })
+                        else:
+                            st.warning("⚠️ No valid response generated. Try a different prompt.")
+                    
+                    else:
+                        # Handle code response
+                        raw_text = data.get("response", "")
+                        generated_code = data.get("generated_code")
+                        explanation = data.get("explanation", "")
+                        
+                        # Fallback: extract code from raw response if not provided
+                        if not generated_code and raw_text:
+                            code_match = re.search(r"```(?:\w+)?\n(.*?)\n```", raw_text, re.DOTALL)
+                            if code_match:
+                                generated_code = code_match.group(1).strip()
+                                if not explanation:
+                                    explanation = raw_text.replace(code_match.group(0), "").strip()
+                        
+                        # If we have explanation or code, add to messages
+                        if explanation or generated_code:
+                            st.session_state.messages.append({
+                                "role": "ai",
+                                "content": (explanation, generated_code)
+                            })
+                        elif raw_text:
+                            # Fallback to raw text
+                            st.session_state.messages.append({
+                                "role": "ai",
+                                "content": raw_text
+                            })
+                        else:
+                            st.warning("⚠️ No valid response generated. Try a different prompt.")
+                    
+                    # Rerun to show the new message
                     st.rerun()
-            else:
-                st.error(f"API Error: {response.status_code} - {response.text}")
-        except requests.exceptions.ConnectionError:
-            st.error("⚠️ Cannot connect to backend! Make sure FastAPI is running.")
+                
+                elif response.status_code == 401:
+                    st.error("🔐 Authentication Error: Invalid API key on backend.")
+                elif response.status_code == 429:
+                    st.error("⏱️ Rate limit exceeded. Please wait a moment and try again.")
+                elif response.status_code == 503:
+                    st.error("🔧 Service temporarily unavailable. Please try again in a moment.")
+                else:
+                    st.error(f"❌ API Error: {response.status_code} - {response.text}")
+            
+            except requests.exceptions.Timeout:
+                st.error("⏱️ Request timed out. The model might be taking too long to respond.")
+            except requests.exceptions.ConnectionError:
+                st.error("⚠️ Cannot connect to backend! Make sure the FastAPI server is running.")
+            except Exception as e:
+                st.error(f"❌ Unexpected error: {str(e)}")
     else:
-        st.warning("Write something in input cell.")
+        st.warning("⚠️ Please enter a prompt before sending.")
+
+# Footer
+st.markdown("---")
+st.markdown(
+    '<div style="text-align: center; color: #888;">Powered by Groq AI | Built with ❤️ using Streamlit</div>',
+    unsafe_allow_html=True
+)