Update app.py

app.py

@@ -80,32 +80,36 @@ def enhance_for_gradio(prompt_text, api_key):
     try:
         result = pipe(
             messages,
-            max_new_tokens=512,
+            max_new_tokens=1024,  # Artırıldı
             temperature=0.7,
             do_sample=True
-            # stop_strings kaldırıldı - test için
         )
         full_response = result[0]["generated_text"]
-
-        # Eğer full_response bir liste ise
+        
+        print("=== MODEL OUTPUT ===")
+        print(full_response)
+        print("=== END OUTPUT ===")
+
+        # Manuel parse denemesi
+        if isinstance(full_response, str):
+            # Kullanıcı prompt'unu bul ve sonrasını al
+            if prompt_text in full_response:
+                idx = full_response.find(prompt_text) + len(prompt_text)
+                enhanced_prompt = full_response[idx:].strip()
+                # <end_of_turn> varsa kaldır
+                enhanced_prompt = enhanced_prompt.replace("<end_of_turn>", "").strip()
+                if enhanced_prompt:
+                    return enhanced_prompt
+        
+        # Veya assistant mesajını bulmaya çalış
         if isinstance(full_response, list):
-            # Assistant rolündeki son mesajı bul
-            assistant_messages = [msg.get("content", "") for msg in full_response if msg.get("role") == "assistant"]
-            if assistant_messages:
-                enhanced_prompt = assistant_messages[-1]
-            else:
-                enhanced_prompt = " ".join([msg.get("content", "") for msg in full_response])
-        else:
-            # String ise direkt kullan
-            enhanced_prompt = full_response
-
-        # Temizlik
-        enhanced_prompt = enhanced_prompt.replace("<end_of_turn>", "").strip()
-
-        if not enhanced_prompt:
-            return "Model did not generate a valid response. Try again."
-
-        return enhanced_prompt
+            for msg in reversed(full_response):
+                if msg.get("role") == "assistant":
+                    content = msg.get("content", "").replace("<end_of_turn>", "").strip()
+                    if content:
+                        return content
+        
+        return "Model did not generate a valid response. Please check the console output."
 
     except Exception as e:
         return f"Enhancement failed: {str(e)}"