Update app.py

app.py

@@ -97,7 +97,7 @@ Do not summarize. Only copy from the context.
 
 }
 
-def generate_prompt_output(prompt_type, query, retriever, llm):
+def generate_output(prompt_type, query, retriever, llm):
     # Try to extract SOL code
     sol_match = re.search(r"\bG\.[A-Z]+\.\d+\b", query)
     matched_code = sol_match.group(0) if sol_match else None