Create app.py

app.py

@@ -0,0 +1,145 @@
+import os
+import io
+import sys
+from dotenv import load_dotenv
+from openai import OpenAI
+import google.generativeai
+import anthropic
+from IPython.display import Markdown, display, update_display
+import gradio as gr
+import subprocess
+
+
+os.environ['OPENAI_API_KEY'] = os.environ.get('OPENAI_API_KEY')
+os.environ['ANTHROPIC_API_KEY'] = os.environ.get('ANTHROPIC_API_KEY')
+deepseek_api_key = os.environ.get('DEEPSEEK_API_KEY')
+)
+
+
+openai = OpenAI()
+claude = anthropic.Anthropic()
+deepseek = OpenAI(
+    api_key=deepseek_api_key, 
+    base_url="https://api.deepseek.com/v1"
+)
+
+OPENAI_MODEL = "gpt-4o"
+CLAUDE_MODEL = "claude-3-5-sonnet-20240620"
+DEEPSEEK_MODEL = 'deepseek-coder'
+
+system_message = "You are an assistant that reimplements Python code in high performance C++ for an M1 Mac. "
+system_message += "Respond only with C++ code; use comments sparingly and do not provide any explanation other than occasional comments. "
+system_message += "The C++ response needs to produce an identical output in the fastest possible time."
+
+
+def user_prompt_for(python):
+    user_prompt = "Rewrite this Python code in C++ with the fastest possible implementation that produces identical output in the least time. "
+    user_prompt += "Respond only with C++ code; do not explain your work other than a few comments. "
+    user_prompt += "Pay attention to number types to ensure no int overflows. Remember to #include all necessary C++ packages such as iomanip.\n\n"
+    user_prompt += python
+    return user_prompt
+
+def messages_for(python):
+    return [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_prompt_for(python)}
+    ]
+# write to a file called optimized.cpp
+
+def write_output(cpp):
+    code = cpp.replace("```cpp","").replace("```","")
+    with open("optimized.cpp", "w") as f:
+        f.write(code)
+
+def stream_gpt(python):    
+    stream = openai.chat.completions.create(model=OPENAI_MODEL, messages=messages_for(python), stream=True)
+    reply = ""
+    for chunk in stream:
+        fragment = chunk.choices[0].delta.content or ""
+        reply += fragment
+        yield reply.replace('```cpp\n','').replace('```','')
+
+def stream_claude(python):
+    result = claude.messages.stream(
+        model=CLAUDE_MODEL,
+        max_tokens=2000,
+        system=system_message,
+        messages=[{"role": "user", "content": user_prompt_for(python)}],
+    )
+    reply = ""
+    with result as stream:
+        for text in stream.text_stream:
+            reply += text
+            yield reply.replace('```cpp\n','').replace('```','')
+
+def stream_deepseek(python):
+    stream = deepseek.chat.completions.create(
+        model = DEEPSEEK_MODEL,
+        messages = messages_for(python),
+        stream=True
+    )
+    reply = ""
+    for chunk in stream:
+        fragment = chunk.choices[0].delta.content or ""
+        reply += fragment
+        yield reply.replace('```cpp\n','').replace('```','')
+
+def optimize(python, model):
+    if model=="GPT":
+        result = stream_gpt(python)
+    elif model=="Claude":
+        result = stream_claude(python)
+    elif model == "DeepSeek":
+        result = stream_deepseek(python)
+    else:
+        raise ValueError("Unknown model")
+    for stream_so_far in result:
+        yield stream_so_far        
+
+def execute_python(code):
+    try:
+        output = io.StringIO()
+        sys.stdout = output
+        exec(code)
+    finally:
+        sys.stdout = sys.__stdout__
+    return output.getvalue()
+
+def execute_cpp(code):
+    write_output(code)
+    try:
+        # Modified compiler flags to be more generic and compatible with Linux environments
+        compile_cmd = ["g++", "-Ofast", "-std=c++17", "-march=native", "-o", "optimized", "optimized.cpp"]
+        compile_result = subprocess.run(compile_cmd, check=True, text=True, capture_output=True)
+        run_cmd = ["./optimized"]
+        run_result = subprocess.run(run_cmd, check=True, text=True, capture_output=True)
+        return run_result.stdout
+    except subprocess.CalledProcessError as e:
+        return f"An error occurred:\n{e.stderr}"
+
+css = """
+.python {background-color: #306998;}
+.cpp {background-color: #050;}
+"""
+
+with gr.Blocks(css=css) as ui:
+    gr.Markdown("## Convert code from Python to C++")
+    with gr.Row():
+        python = gr.Textbox(label="Python code:", value=python_hard, lines=10)
+        cpp = gr.Textbox(label="C++ code:", lines=10)
+    with gr.Row():
+        model = gr.Dropdown(["GPT", "Claude","DeepSeek"], label="Select model", value="Claude")
+    with gr.Row():
+        convert = gr.Button("Convert code")
+    with gr.Row():
+        python_run = gr.Button("Run Python")
+        cpp_run = gr.Button("Run C++")
+    with gr.Row():
+        python_out = gr.TextArea(label="Python result:", elem_classes=["python"])
+        cpp_out = gr.TextArea(label="C++ result:", elem_classes=["cpp"])
+
+    convert.click(optimize, inputs=[python, model], outputs=[cpp])
+    python_run.click(execute_python, inputs=[python], outputs=[python_out])
+    cpp_run.click(execute_cpp, inputs=[cpp], outputs=[cpp_out])
+
+ui.launch(inbrowser=True)