Update src/streamlit_app.py

src/streamlit_app.py

@@ -161,7 +161,7 @@ def search_index(query: str, index, meta, top_k: int, threshold: float) -> List[
     return results
 
 def generate_answer(context: str, query: str) -> str:
-    """Generate robust answer with explicit citations."""
+    """Generate robust answer with explicit citations — auto-switches between endpoints."""
     prompt = f"""
 You are a precise academic assistant specialized in university policies.
 Use only the provided *USTP Student Handbook 2023 Edition* content as reference.
@@ -181,6 +181,10 @@ If the answer is not explicitly found, respond with:
 - Do NOT invent or assume any information not in the handbook.
 """
 
+    if not hf_client:
+        return "❌ Hugging Face client not initialized."
+
+    # Try standard text-generation first
     try:
         response = hf_client.text_generation(
             model=DEFAULT_MODEL,
@@ -190,8 +194,21 @@ If the answer is not explicitly found, respond with:
             repetition_penalty=1.1,
         )
         return response
-    except Exception as e:
-        return f"⚠️ Error generating answer: {e}"
+    except Exception as e1:
+        # If it fails, automatically switch to conversational API
+        try:
+            chat_response = hf_client.chat.completions.create(
+                model=DEFAULT_MODEL,
+                messages=[
+                    {"role": "system", "content": "You are a precise and factual handbook assistant."},
+                    {"role": "user", "content": prompt},
+                ],
+                max_tokens=400,
+                temperature=0.25,
+            )
+            return chat_response.choices[0].message["content"]
+        except Exception as e2:
+            return f"⚠️ Error generating answer: {e2}"
 
 # =============================================================
 # 🔍 Index Handling