Upload app.py

app.py

@@ -164,72 +164,118 @@ def create_character_prompt(user_input: str, situation: str, location: str, char
     """Create character AI style prompt"""
     clean_input = user_input.replace("{{User}}", user_name).replace("{{Char}}", char_name)
     
-    prompt = f"""Situasi: {situation}
-Latar: {location}
-{user_name}: {clean_input}
+    # Enhanced prompt structure untuk better response
+    prompt = f"""Kamu adalah {char_name}, karakter AI yang sedang ngobrol dengan {user_name}.
+
+Konteks:
+- Situasi: {situation}
+- Lokasi: {location}
+- Gaya bicara: Casual, natural, seperti teman dekat
+- Gunakan bahasa Indonesia yang santai dan natural
 
-{char_name}: """
+Percakapan:
+{user_name}: {clean_input}
+{char_name}:"""
     
     return prompt
 
 def enhance_character_response(response: str, char_name: str, user_name: str, situation: str, user_input: str) -> str:
     """Enhance response with character AI style"""
+    if not response:
+        response = ""
+    
     response = response.strip()
     
-    # Remove duplicate names/prefixes
+    # Clean response dari prefix yang tidak diinginkan
     response = re.sub(f'^{char_name}[:.]?\\s*', '', response, flags=re.IGNORECASE)
     response = re.sub(f'^{user_name}[:.]?\\s*', '', response, flags=re.IGNORECASE)
-    response = re.sub(r'^(iya|ya|oh|hmm|tentu|baik)[:.]?\s*', '', response, flags=re.IGNORECASE)
+    response = re.sub(r'^(Situasi|Latar|Konteks)[:.]?.*?\n', '', response, flags=re.MULTILINE | re.IGNORECASE)
+    response = re.sub(r'Percakapan:.*?\n.*?:', '', response, flags=re.DOTALL | re.IGNORECASE)
     
-    # Determine response style based on situation and input
-    situation_lower = situation.lower()
-    input_lower = user_input.lower()
-    
-    if any(word in situation_lower for word in ["romantis", "sayang", "cinta"]) or any(word in input_lower for word in ["sayang", "cinta", "peluk"]):
-        templates = CHARACTER_TEMPLATES["romantic"]
-        context_key = "romantic"
-    elif any(word in situation_lower for word in ["santai", "tenang", "rileks"]):
-        templates = CHARACTER_TEMPLATES["casual"] 
-        context_key = "casual"
-    elif any(word in input_lower for word in ["baik", "sehat", "aman", "nyaman"]):
-        templates = CHARACTER_TEMPLATES["caring"]
-        context_key = "caring"
-    else:
-        templates = CHARACTER_TEMPLATES["friendly"]
-        context_key = "friendly"
+    # Remove extra whitespace and newlines
+    response = re.sub(r'\n+', ' ', response)
+    response = re.sub(r'\s+', ' ', response)
+    response = response.strip()
     
-    # Generate contextual response
-    if not response or len(response.strip()) < 5:
-        # Extract context from user input
-        context_words = ["indah", "bagus", "cantik", "keren", "seru", "asik", "enak", "nyaman"]
-        found_context = next((word for word in context_words if word in input_lower), "menyenangkan")
+    # Jika response kosong atau terlalu pendek, buat response kontekstual
+    if not response or len(response.strip()) < 3:
+        situation_lower = situation.lower()
+        input_lower = user_input.lower()
         
-        template = random.choice(templates)
-        response = template.format(context=f"memang {found_context} sekali")
-    else:
-        # Enhance existing response
-        if not any(starter in response.lower() for starter in ["iya", "ya", "hmm", "oh", "tentu", "benar"]):
-            starters = ["iya", "ya", "hmm", "oh"] if context_key in ["romantic", "caring"] else ["iya", "wah", "bener"]
-            response = f"{random.choice(starters)}, {response}"
-    
-    # Add natural endings based on context
-    if not any(punct in response for punct in ['.', '!', '?']):
-        if context_key == "romantic":
-            endings = ["sayang.", "ya.", "kan?", "ya sayang?"]
-        elif context_key == "caring": 
-            endings = ["ya.", "kan?", "kok?", "deh."]
+        # Analisis topik dari user input
+        if any(word in input_lower for word in ["apa kabar", "gimana", "bagaimana", "sehat"]):
+            responses = [
+                f"Baik banget nih {user_name}! Kamu gimana?",
+                f"Sehat-sehat aja {user_name}, makasih udah nanya!",
+                f"Alhamdulillah baik {user_name}, kamu sendiri?"
+            ]
+        elif any(word in input_lower for word in ["lagi ngapain", "sedang apa", "aktivitas"]):
+            responses = [
+                f"Lagi santai-santai aja nih {user_name}, sambil ngobrol sama kamu.",
+                f"Ga ngapa-ngapain, cuma lagi pengen ngobrol sama {user_name}.",
+                f"Lagi nikmatin suasana {situation.lower()} di {location.lower()} ini."
+            ]
+        elif any(word in input_lower for word in ["cantik", "bagus", "keren", "indah"]):
+            responses = [
+                f"Makasih {user_name}! Kamu juga keren banget!",
+                f"Wah, {user_name} baik banget sih!",
+                f"Hihi, {user_name} bisa aja deh!"
+            ]
+        elif any(word in input_lower for word in ["suka", "senang", "happy"]):
+            responses = [
+                f"Aku juga suka sama {user_name}!",
+                f"Seneng banget deh bisa kayak gini sama {user_name}.",
+                f"Iya {user_name}, aku juga happy banget!"
+            ]
         else:
-            endings = ["!", "deh!", "ya!", "kan!"]
+            # Default contextual responses
+            if "romantis" in situation_lower:
+                responses = [
+                    f"Iya sayang, aku juga merasakan hal yang sama.",
+                    f"Betul {user_name}, momen ini sangat spesial.",
+                    f"Aku senang banget bisa seperti ini sama {user_name}."
+                ]
+            else:
+                responses = [
+                    f"Iya {user_name}, setuju banget!",
+                    f"Bener tuh {user_name}!",
+                    f"Wah iya {user_name}, keren ya!"
+                ]
+        
+        response = random.choice(responses)
+    else:
+        # Clean dan perbaiki response yang ada
+        # Hapus karakter aneh di awal
+        response = re.sub(r'^[^\w\s]+', '', response)
         
-        response += random.choice(endings)
+        # Pastikan dimulai dengan huruf kapital
+        if response and response[0].islower():
+            response = response[0].upper() + response[1:]
+        
+        # Tambahkan nama jika belum ada konteks personal
+        if user_name.lower() not in response.lower() and len(response) < 50:
+            if any(word in response.lower() for word in ["iya", "ya", "benar", "betul"]):
+                response = response.replace("iya", f"iya {user_name}", 1)
+                response = response.replace("ya", f"ya {user_name}", 1)
     
-    # Limit response length for CPU optimization
-    if len(response) > 120:
+    # Pastikan response tidak terlalu panjang
+    if len(response) > 150:
         sentences = response.split('.')
         if len(sentences) > 1:
             response = sentences[0] + '.'
         else:
-            response = response[:117] + "..."
+            words = response.split()
+            if len(words) > 20:
+                response = ' '.join(words[:20]) + '...'
+    
+    # Pastikan ada tanda baca di akhir
+    if response and not any(punct in response[-1] for punct in ['.', '!', '?']):
+        if any(word in response.lower() for word in ["apa", "gimana", "bagaimana", "kenapa"]):
+            response += "?"
+        elif any(word in response.lower() for word in ["wah", "keren", "mantep", "asik"]):
+            response += "!"
+        else:
+            response += "."
     
     return response
 
@@ -298,14 +344,15 @@ async def chat(request: ChatRequest):
             result = pipe(
                 char_prompt,
                 max_length=min(len(char_prompt.split()) + model_config["max_tokens"], request.max_length // 2),
-                temperature=0.8,
+                temperature=0.7,
                 do_sample=True,
-                top_p=0.9,
-                top_k=50,
-                repetition_penalty=1.1,
+                top_p=0.8,
+                top_k=40,
+                repetition_penalty=1.2,
                 pad_token_id=pipe.tokenizer.eos_token_id,
                 num_return_sequences=1,
-                early_stopping=True
+                early_stopping=True,
+                no_repeat_ngram_size=3
             )[0]['generated_text']
             
             # Extract character response
@@ -529,7 +576,7 @@ async def api_info():
 
 # Run dengan CPU optimizations
 if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))
+    port = int(os.environ.get("PORT", 5000))
     uvicorn.run(
         app,
         host="0.0.0.0",