Update app.py

app.py

@@ -48,33 +48,55 @@ from dateutil.parser import parse
 # Local modules
 from footer import footer
 
-load_dotenv() 
-llm = Together(
-    model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-    temperature=0.7,
-    max_tokens=1024,
-    together_api_key=os.getenv('TOGETHER_API_KEY')
-)
-
+# ----------------- Streamlit Config -------------------
+st.set_page_config(page_title="ApnaLawyer", layout="centered")
 
-# Load environment variables
+# ----------------- Load Environment Variables -------------------
 load_dotenv()
 SPEECHMATICS_API_KEY = os.getenv('SPEECHMATICS_API_KEY')
-api_key = os.getenv('TOGETHER_API_KEY')  # Set this in your environment
-if not api_key:
+TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
+
+if not TOGETHER_API_KEY:
     st.error("Please set TOGETHER_API_KEY environment variable")
     st.stop()
 
+# ----------------- LLM Init -------------------
+llm = Together(
+    model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+    temperature=0.7,
+    max_tokens=1024,
+    together_api_key=TOGETHER_API_KEY
+)
+
+# ----------------- Firebase Init (Pyrebase) -------------------
+firebase_config = {
+    "apiKey": os.environ["FIREBASE_API_KEY"],
+    "authDomain": os.environ["FIREBASE_AUTH_DOMAIN"],
+    "databaseURL": os.environ["FIREBASE_DATABASE_URL"],
+    "projectId": os.environ["FIREBASE_PROJECT_ID"],
+    "storageBucket": os.environ["FIREBASE_STORAGE_BUCKET"],
+    "messagingSenderId": os.environ["FIREBASE_MESSAGING_SENDER_ID"],
+    "appId": os.environ["FIREBASE_APP_ID"]
+}
 
+firebase = pyrebase.initialize_app(firebase_config)
+auth = firebase.auth()
 
+# ----------------- Firebase Admin SDK Init -------------------
+def initialize_firebase():
+    try:
+        if not firebase_admin._apps:
+            creds_dict = json.loads(os.environ["FIREBASE_CREDS_JSON"])
+            cred = credentials.Certificate(creds_dict)
+            firebase_app = initialize_app(cred, {
+                'databaseURL': os.environ["FIREBASE_DB_URL"]
+            })
+    except Exception as e:
         st.error(f"Firebase initialization error: {str(e)}")
 
 initialize_firebase()
 
-
-# ----------------- Streamlit Config -------------------
-st.set_page_config(page_title="ApnaLawyer", layout="centered")
-
+# ----------------- UI -------------------
 col1, col2, col3 = st.columns([1, 30, 1])
 with col2:
     st.image("images/banner.jpg", use_container_width=True)
@@ -86,47 +108,28 @@ st.markdown("""
     </style>
     """, unsafe_allow_html=True)
 
-
-
-
-
-# Function to translate text
-def translate_text(text, target_language):
-    try:
-        # Use GoogleTranslator from deep-translator
-        translated_text = GoogleTranslator(source='auto', target=target_language).translate(text)
-        return translated_text
-    except Exception as e:
-        return f"⚠ Translation failed: {str(e)}"
-    
-
+# ----------------- Translation Logic -------------------
 supported_languages = {
     "hindi": "hi",
     "english": "en",
-    "hinglish": "hi"  # special handling below
+    "hinglish": "hi"
 }
 
 devanagari_regex = re.compile(r'[\u0900-\u097F]+')
 
-
 def detect_target_language(prompt):
-    """Detect if the response should be in Hindi, Hinglish, or English only."""
     prompt_lower = prompt.lower().strip()
 
-    # Block Kannada and other unsupported scripts
-    if re.search(r'[\u0C80-\u0CFF]', prompt):  # Kannada unicode block
+    if re.search(r'[\u0C80-\u0CFF]', prompt):  # Kannada block
         return "hindi"
 
-    # Check if explicitly mentioned like: 'in hindi'
     for lang_name, lang_code in supported_languages.items():
         if f"in {lang_name}" in prompt_lower:
             return lang_name
 
-    # If it contains Devanagari, assume Hindi
     if devanagari_regex.search(prompt):
         return "hindi"
 
-    # Detect Hinglish by common Hindi terms in Latin script
     if re.search(r'\bdhara\b|\bkanoon\b|\bnyay\b', prompt_lower) and detect(prompt) == 'en':
         return "hinglish"
 
@@ -140,37 +143,11 @@ def detect_target_language(prompt):
 
     return "english"
 
-
-# Initialize the translator
-#translator = Translator()
-
-# ----------------- Firebase Init (Pyrebase) -------------------
-firebase_config = {
-    "apiKey": os.environ["FIREBASE_API_KEY"],
-    "authDomain": os.environ["FIREBASE_AUTH_DOMAIN"],
-    "databaseURL": os.environ["FIREBASE_DATABASE_URL"],  # Updated to match your variable
-    "projectId": os.environ["FIREBASE_PROJECT_ID"],
-    "storageBucket": os.environ["FIREBASE_STORAGE_BUCKET"],
-    "messagingSenderId": os.environ["FIREBASE_MESSAGING_SENDER_ID"],
-    "appId": os.environ["FIREBASE_APP_ID"]
-}
-
-firebase = pyrebase.initialize_app(firebase_config)
-auth = firebase.auth()
-
-# ----------------- Firebase Admin SDK Init -------------------
-def initialize_firebase():
+def translate_text(text, target_language):
     try:
-        if not firebase_admin._apps:
-            # Load JSON from environment variable
-            creds_dict = json.loads(os.environ["FIREBASE_CREDS_JSON"])
-            cred = credentials.Certificate(creds_dict)
-            firebase_app = initialize_app(cred, {
-                'databaseURL': os.environ["FIREBASE_DB_URL"]
-            })
+        return GoogleTranslator(source='auto', target=target_language).translate(text)
     except Exception as e:
-        st.error(f"Firebase initialization error: {str(e)}")
-
+        return f"⚠ Translation failed: {str(e)}"
 
 # ----------------- Login/Signup Interface -------------------
 import json