| 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() |
|
|
