import gradio as gr
import openai
import pyttsx3

openai.api_key = "YOUR_OPENAI_KEY"

# Text-to-Speech engine
engine = pyttsx3.init()

def ai_assistant(audio):
    # Convert speech → text
    import speech_recognition as sr
    recognizer = sr.Recognizer()
    with sr.AudioFile(audio) as source:
        audio_data = recognizer.record(source)
        transcript = recognizer.recognize_google(audio_data)

    # Send to AI
    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[{"role": "user", "content": transcript}]
    )
    reply = response.choices[0].message["content"]

    # Convert AI reply → speech
    engine.save_to_file(reply, "response.wav")
    engine.runAndWait()

    return reply, "response.wav"

app = gr.Interface(
    fn=ai_assistant,
    inputs=gr.Audio(source="microphone", type="filepath"),
    outputs=[gr.Textbox(label="AI Response"), gr.Audio(type="filepath")],
    live=True,
    title="🎙️ AI Coding Assistant"
)

app.launch()