Update app.py

app.py

@@ -38,7 +38,40 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 # Database configuration
 DATABASE_PATH = '/tmp/chat_database.db'
 
-# Database connection manager
+# Initialize prompt template
+prompt_template = """
+Role: You are Figr Code Assistant, specializing in providing clear, error-free Python code solutions.
+
+Context:
+{important_info}
+
+Previous Conversation:
+{chat_history}
+
+Current Request:
+{user_request}
+
+Output Guidelines:
+1. Code Format:
+   - Use ```python for code blocks
+   - Use `code` for inline code references
+   - Provide raw text without HTML formatting
+   - Strictly include explanation only after code blocks
+
+2. Code Organization:
+   - Default to single, focused code snippets for clarity
+   - Only split into multiple snippets(each individually runnable) if:
+     a) Multiple distinct concepts are requested
+     b) Complex functionality requires modular explanation
+     
+   - Mark critical information with [IMPORTANT] prefix and give small explanations with some bold headings if required and in white font always.
+"""
+
+prompt = PromptTemplate(
+    input_variables=["user_request", "chat_history", "important_info"],
+    template=prompt_template
+)
+
 @contextmanager
 def get_db_connection():
     """Context manager for database connections"""
@@ -666,6 +699,10 @@ try:
     print("Starting model initialization...")
     llm = initialize_model()
     print("Model initialized successfully")
+    
+    print("Creating LLM chain...")
+    llm_chain = LLMChain(llm=llm, prompt=prompt)
+    print("LLM chain created successfully")
 except Exception as e:
     print(f"Initialization error: {e}")
     raise