Update app.py

app.py

@@ -52,13 +52,10 @@ if query:
             response = llm([{"role": "system", "content": "You are a helpful assistant."},
                             {"role": "user", "content": prompt}])
 
-            # استخراج پاسخ مدل
-            if isinstance(response, dict) and 'choices' in response:
-                rewritten = response['choices'][0]['message']['content'].strip()
-                st.markdown("🎨 **بازنویسی شده با LLM:**")
-                st.write(rewritten)
-            else:
-                st.warning("پاسخ دریافتی معتبر نیست.")
+            rewritten = response.strip()
+            st.markdown("🎨 **بازنویسی شده با LLM:**")
+            st.write(rewritten)
+            st.warning("پاسخ دریافتی معتبر نیست.")
 
             found = True
             break