|
|
import os |
|
|
import openai |
|
|
import gradio as gr |
|
|
|
|
|
|
|
|
client = openai.OpenAI( |
|
|
api_key=os.environ.get("SAMBANOVA_API_KEY"), |
|
|
base_url="https://api.sambanova.ai/v1", |
|
|
) |
|
|
|
|
|
def generate_haiku_comment(code): |
|
|
""" |
|
|
Generates a haiku comment for the given code snippet. |
|
|
""" |
|
|
prompt = ( |
|
|
f"Analyze the following code and perform two tasks:\n\n" |
|
|
f"1. Write a poetic haiku as a comment summarizing its purpose or essence.\n" |
|
|
f"2. Refactor and clean the code while maintaining its functionality.\n\n" |
|
|
f"Code:\n{code}\n\n" |
|
|
"Provide the output as haiku comment followed by the refactored code." |
|
|
) |
|
|
|
|
|
try: |
|
|
response = client.chat.completions.create( |
|
|
model='Qwen2.5-Coder-32B-Instruct', |
|
|
messages=[ |
|
|
{"role": "system", "content": "You are a creative haiku generator and code refactoring assistant."}, |
|
|
{"role": "user", "content": prompt} |
|
|
], |
|
|
temperature=0.7, |
|
|
top_p=0.9 |
|
|
) |
|
|
haiku_comment = response.choices[0].message.content.strip() |
|
|
return haiku_comment |
|
|
except Exception as e: |
|
|
return f"Error generating haiku comment: {str(e)}" |
|
|
|
|
|
|
|
|
def gradio_interface(): |
|
|
interface = gr.Interface( |
|
|
fn=generate_haiku_comment, |
|
|
inputs=gr.Textbox(label="Code Snippet",lines=10, placeholder="Paste your code snippet here..."), |
|
|
outputs= gr.Textbox(label="Code will be generated herep",lines=10), |
|
|
title="Code Haiku Comment Generator", |
|
|
description="Paste a code snippet, and let the AI generate a poetic haiku comment summarizing it!", |
|
|
) |
|
|
return interface |
|
|
|
|
|
if __name__ == "__main__": |
|
|
gradio_interface().launch() |
|
|
|
