Create app.py

app.py

@@ -0,0 +1,207 @@
+import gradio as gr
+import openai
+import os
+import tempfile
+import sys
+import io
+from contextlib import redirect_stdout
+
+# Use Together AI API (OpenAI-compatible) - Sign up at https://www.together.ai/ for free API key
+# Models available: CodeLlama, StarCoder, etc. (lightweight, no local torch needed)
+TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")  # Set in Hugging Face Spaces secrets
+client = openai.OpenAI(
+    api_key=TOGETHER_API_KEY,
+    base_url="https://openrouter.ai/api/v1"
+)
+
+# Model: A code-specialized model from Together AI (e.g., CodeLlama 7B Instruct - efficient and accurate for Python)
+MODEL_NAME = "openai/gpt-oss-20b:free"  # Alternative: "Salesforce/codegen-350M-mono" for lighter
+
+def generate_code_with_together(instruction, file_paths):
+    """
+    Generate Python code using Together AI API (OpenAI-compatible).
+    """
+    prompt = f"""
+You are a Python expert. The user has provided the instruction: "{instruction}"
+And input files: {file_paths} (these paths are available in the current working directory).
+
+Write a complete, self-contained Python script that:
+1. Reads the input file(s) from the given paths.
+2. Performs the requested operation (e.g., for "remove background from image", use rembg or similar).
+3. Saves the output file(s) to a temporary directory (use tempfile.mkdtemp() for a temp dir).
+4. At the very end of the script, print exactly one line: OUTPUT_FILE_PATH: /full/path/to/output/file.ext
+   (Replace with the actual full path to the generated output file. If multiple outputs, choose the main one.)
+
+Use only safe, standard libraries or pre-installed ones like PIL (from pillow), rembg for image tasks, numpy, etc.
+Do not use any network calls, file deletions outside temp, or unsafe operations.
+Handle errors gracefully and assume the environment has the necessary libs (e.g., import rembg).
+
+Example for background removal:
+from rembg import remove
+import os
+import tempfile
+input_path = '{file_paths[0]}'
+temp_dir = tempfile.mkdtemp()
+with open(input_path, 'rb') as i:
+    input_data = i.read()
+    output_data = remove(input_data)
+    output_path = os.path.join(temp_dir, 'output.png')
+    with open(output_path, 'wb') as o:
+        o.write(output_data)
+print(f"OUTPUT_FILE_PATH: {{output_path}}")
+
+Generate ONLY the Python code. Do not add explanations.
+"""
+    
+    try:
+        response = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=[
+                {"role": "system", "content": "You are a helpful Python code generator. Respond only with clean, executable Python code."},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=1000,
+            temperature=0.2,
+            stop=["```"]  # Stop at code block end if any
+        )
+        
+        generated_code = response.choices[0].message.content.strip()
+        
+        # Clean up: Remove any markdown code blocks (e.g., ```python ... ```)
+        if generated_code.startswith("```python"):
+            generated_code = generated_code[10:].strip()
+        if generated_code.endswith("```"):
+            generated_code = generated_code[:-3].strip()
+        
+        return generated_code
+        
+    except Exception as api_error:
+        raise api_error
+
+def process_request(instruction, files):
+    if not files:
+        return "لطفاً حداقل یک فایل آپلود کنید.", None
+    
+    file_paths = [file.name for file in files]
+    
+    try:
+        # Generate code using Together AI
+        generated_code = generate_code_with_together(instruction, file_paths)
+        
+        if not generated_code or len(generated_code) < 50:
+            return f"کد مناسبی تولید نشد. پرامپت: {instruction}\nکد تولید شده:\n{generated_code}", None
+        
+        # Create a temporary file for the generated code
+        with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as temp_code_file:
+            temp_code_file.write(generated_code)
+            code_file_path = temp_code_file.name
+        
+        # Capture the output of executing the generated code
+        f = io.StringIO()
+        output_path = None
+        
+        with redirect_stdout(f):
+            try:
+                # Execute the generated code in a restricted environment (for safety)
+                safe_globals = {
+                    '__builtins__': {
+                        'print': print,
+                        'open': open,
+                        'range': range,
+                        'len': len,
+                        'str': str,
+                        'int': int,
+                        'float': float,
+                        'list': list,
+                        'dict': dict,
+                        'tuple': tuple,
+                        'set': set,
+                        'bool': bool,
+                        'type': type,
+                        'isinstance': isinstance,
+                        'hasattr': hasattr,
+                        'getattr': getattr,
+                        'setattr': setattr,
+                        'delattr': delattr,
+                        'abs': abs,
+                        'all': all,
+                        'any': any,
+                        'bin': bin,
+                        'chr': chr,
+                        'complex': complex,
+                        'divmod': divmod,
+                        'enumerate': enumerate,
+                        'format': format,
+                        'hash': hash,
+                        'hex': hex,
+                        'id': id,
+                        'iter': iter,
+                        'max': max,
+                        'min': min,
+                        'next': next,
+                        'oct': oct,
+                        'ord': ord,
+                        'pow': pow,
+                        'repr': repr,
+                        'round': round,
+                        'slice': slice,
+                        'sorted': sorted,
+                        'sum': sum,
+                        'zip': zip,
+                        '__import__': __import__,
+                    },
+                    'os': os,
+                    'tempfile': tempfile,
+                    'sys': sys,
+                    'PIL': __import__('PIL', fromlist=['Image']),
+                    'rembg': __import__('rembg', fromlist=['remove']),
+                    # Add more safe imports as needed (e.g., 'numpy': __import__('numpy'))
+                }
+                safe_locals = {}
+                
+                exec(generated_code, safe_globals, safe_locals)
+                
+                # Capture the printed OUTPUT_FILE_PATH
+                stdout_output = f.getvalue()
+                if "OUTPUT_FILE_PATH:" in stdout_output:
+                    output_path = stdout_output.split("OUTPUT_FILE_PATH:")[-1].strip()
+                    output_path = output_path.split('\n')[0].strip()
+                
+                if not output_path and 'output_path' in safe_locals:
+                    output_path = safe_locals.get('output_path')
+                    
+            except Exception as exec_error:
+                return f"خطا در اجرای کد تولید شده: {str(exec_error)}\nکد تولید شده:\n{generated_code}", None
+        
+        # Clean up
+        os.unlink(code_file_path)
+        
+        if output_path and os.path.exists(output_path):
+            return f"عملیات با موفقیت انجام شد!\nکد تولید شده توسط Together AI:\n{generated_code}\n\nخروجی stdout:\n{stdout_output}", output_path
+        else:
+            return f"فایل خروجی یافت نشد. خروجی:\n{stdout_output}\nکد:\n{generated_code}", None
+            
+    except Exception as gen_error:
+        return f"خطا در تولید کد (Together AI): {str(gen_error)}\nپرامپت:\n{instruction}", None
+
+# Gradio Interface
+with gr.Blocks(title="AI File Processor - Together AI") as demo:
+    gr.Markdown("# پردازشگر فایل با هوش مصنوعی (Together AI)\n\nدستور خود را وارد کنید و فایل(ها) را آپلود کنید. API Together AI (سازگار با OpenAI) کد پایتون می‌نویسد و اجرا می‌کند.\n\n**مدل استفاده شده:** CodeLlama-7B-Instruct\n**API Key:** در Secrets Hugging Face با نام TOGETHER_API_KEY ست کنید.")
+    
+    with gr.Row():
+        instruction_input = gr.Textbox(label="دستور (مثال: بک گراند عکس را حذف کن)", placeholder="دستور خود را بنویسید...")
+        files_input = gr.File(file_count="multiple", label="فایل(ها)")
+    
+    process_btn = gr.Button("پردازش با Together AI")
+    
+    output_text = gr.Textbox(label="نتیجه و کد", lines=10)
+    output_file = gr.File(label="فایل خروجی (لینک دانلود)")
+    
+    process_btn.click(
+        fn=process_request,
+        inputs=[instruction_input, files_input],
+        outputs=[output_text, output_file]
+    )
+
+if __name__ == "__main__":
+    demo.launch(share=True)