Update app.py

app.py

@@ -17,6 +17,14 @@ from docx import Document
 from gtts import gTTS
 from io import BytesIO
 import spacy
+import subprocess
+
+# -----------------------------
+# Hugging Face fix: ensure Streamlit runs properly
+# -----------------------------
+#if _name_ == "_main_" and os.environ.get("SYSTEM") == "spaces":
+  #  subprocess.Popen(["streamlit", "run", "app.py", "--server.port", "7860", "--server.address", "0.0.0.0"])
+   # exit()
 
 # -----------------------------
 # Page config
@@ -59,10 +67,9 @@ def load_models():
 
     return tokenizer_simplify, simplify_model, gen_tokenizer, gen_model, nlp, classifier, summarizer
 
-tokenizer_simplify, simplify_model, gen_tokenizer, gen_model, nlp, classifier, summarizer = load_models()
 
-# Hugging Face Spaces likely CPU-only
-DEVICE = "cpu"
+tokenizer_simplify, simplify_model, gen_tokenizer, gen_model, nlp, classifier, summarizer = load_models()
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 gen_model.to(DEVICE)
 
 # -----------------------------
@@ -154,7 +161,8 @@ def fairness_score_visual(text, lang):
 
 def chat_response(prompt, lang, history):
     """Persistent memory chat"""
-    context = "\n".join([f"User: {u}\nAI: {a}" for u, a in history[-3:]])  # Last 3 turns
+    # Combine chat history context
+    context = "\n".join([f"User: {u}\nAI: {a}" for u, a in history[-3:]])  # Keep last 3
     full_prompt = f"You are a helpful multilingual legal assistant. {context}\nUser: {prompt}\nAI:"
     inputs = gen_tokenizer(full_prompt, return_tensors="pt").to(DEVICE)
     outputs = gen_model.generate(**inputs, max_new_tokens=200, temperature=0.7, top_p=0.9, do_sample=True)
@@ -163,6 +171,7 @@ def chat_response(prompt, lang, history):
         response = response.split("AI:")[-1].strip()
     return translate_text(response, lang)
 
+
 # -----------------------------
 # Main Streamlit App
 # -----------------------------
@@ -214,6 +223,7 @@ def main():
         lang = st.selectbox("Chat Language:", LANG_NAMES, index=0)
         query = st.text_area("Ask your question:", height=150)
 
+        # Maintain persistent conversation
         if "chat_history" not in st.session_state:
             st.session_state.chat_history = []
 
@@ -227,6 +237,7 @@ def main():
                     if audio:
                         st.audio(audio, format="audio/mp3")
 
+        # Display conversation history
         if st.session_state.chat_history:
             st.markdown("### 🧠 Chat History")
             for q, a in st.session_state.chat_history[-5:]:
@@ -249,5 +260,6 @@ def main():
         *Disclaimer:* Educational use only — not legal advice.
         """)
 
-if __name__ == "__main__":
-    main()
+
+if _name_ == "_main_":
+    main()