Update app.py

app.py

@@ -36,33 +36,19 @@ Choose the most appropriate response structure (1-5) and generate the response a
     def generate_response(self, context, query, chat_history=''):
         try:
             # Generate structured response
-            response = self.response_chain.run({
+            full_response = self.response_chain.run({
                 'context': context,
                 'query': query,
                 'chat_history': chat_history or "No previous context"
             })
             
-            # Parse the response to extract structure and content
-            structure_choice = int(response[0]) if response[0].isdigit() else 1
-            response_content = response[1:].strip()
+            # Extract only the actual response content (after the structure selection)
+            response_content = full_response[1:].strip()
             
-            return self._format_response(structure_choice, response_content)
+            return response_content
         except Exception as e:
             logging.error(f"Response generation error: {e}")
-            return self._default_response(query)
-
-    def _format_response(self, structure_choice, content):
-        structures = {
-            1: f"## Technical Breakdown\n{content}",
-            2: f"📍 Key Insights:\n{content}",
-            3: f"### Structured Insights\n{content}",
-            4: f"🔍 Narrative Explanation:\n{content}",
-            5: f"🔬 Comparative Analysis:\n{content}"
-        }
-        return structures.get(structure_choice, structures[1])
-
-    def _default_response(self, query):
-        return f"I couldn't generate a structured response for: {query}"
+            return f"I couldn't generate a response for: {query}"
 
 class AdvancedPdfChatbot:
     def __init__(self, openai_api_key):