Update app.py

app.py

@@ -1,18 +1,16 @@
-import os
 import openai
 import gradio as gr
 
-# بهتر است API key را امن نگه دارید (در تست‌ها استفاده شود فقط)
-openai.api_key = os.getenv("OPENAI_API_KEY", "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")
-client = openai.OpenAI(api_key=openai.api_key)
-
-def translate_text(text, target_language, context):
+def translate_text(api_key, text, target_language, context):
+    if not api_key:
+        return "⚠️ Please enter your OpenAI API key."
     if not text or not target_language:
-        return "Please enter text and choose language."
-
-    prompt = f"Given the context: {context}\nTranslate to {target_language}:\n{text}" if context else f"Translate to {target_language}:\n{text}"
+        return "⚠️ Please enter text and choose language."
 
     try:
+        client = openai.OpenAI(api_key=api_key)
+        prompt = f"Given the context: {context}\nTranslate to {target_language}:\n{text}" if context else f"Translate to {target_language}:\n{text}"
+
         response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
@@ -21,21 +19,22 @@ def translate_text(text, target_language, context):
             ]
         )
         return response.choices[0].message.content.strip()
+
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
 iface = gr.Interface(
     fn=translate_text,
     inputs=[
-        gr.Textbox(label="Text to translate"),
-        gr.Dropdown(["Persian", "English"], label="Target Language"),
-        gr.Textbox(label="Context (optional)")  # حذف optional=True
+        gr.Textbox(label="🔑 OpenAI API Key", type="password", placeholder="sk-..."),
+        gr.Textbox(label="📝 Text to translate"),
+        gr.Dropdown(["Persian", "English"], label="🌐 Target Language"),
+        gr.Textbox(label="📚 Context (optional)", placeholder="مثلاً زمینه ترجمه: متنی رسمی، چت دوستانه و غیره")
     ],
     outputs="text",
-    title="AI Translation Agent",
-    description="ترجمه با استفاده از GPT-3.5 و context دلخواه شما"
+    title="🤖 AI Translation Agent",
+    description="🌍 ترجمه با استفاده از GPT-3.5 با پشتیبانی از زمینه دلخواه \n\nبرای استفاده، کلید OpenAI API خود را وارد کنید."
 )
 
 if __name__ == "__main__":
-    iface.launch()
-    
+    iface.launch()